winidows下安装pytorch报PackageNotFoundError:cudatoolkit错误的解决方法

今天给新电脑装pytorch的时候查到MX450驱动的CUDA版本是11.1,于是兴冲冲跑去pytorch官网找到安装命令准备安装

winidows下安装pytorch报PackageNotFoundError:cudatoolkit错误的解决方法_第1张图片

pytorch官网告诉我命令是这个

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia

还特别叮嘱我:'nvidia' channel is required for cudatoolkit 11.1,结果执行就报错如下

winidows下安装pytorch报PackageNotFoundError:cudatoolkit错误的解决方法_第2张图片

加了清华源下载还是没用,然后最后去anaconda官网一搜,找到原因了,原因是在在指定的nvidia通道中(-c nvidia)只有linux-64版本的包,没有windows的,官网命令把我给坑了

winidows下安装pytorch报PackageNotFoundError:cudatoolkit错误的解决方法_第3张图片

将命令改为下面之后,问题解决

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge

winidows下安装pytorch报PackageNotFoundError:cudatoolkit错误的解决方法_第4张图片

你可能感兴趣的:(环境配置,pytorch,机器学习,anaconda)