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

pytorch官网告诉我命令是这个
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia
还特别叮嘱我:'nvidia' channel is required for cudatoolkit 11.1,结果执行就报错如下

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

将命令改为下面之后,问题解决
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge

本文记录了一次安装PyTorch过程中遇到的问题及解决方案。作者尝试在新电脑上安装带有CUDA 11.1支持的PyTorch时,依据官方指南操作却遭遇错误。经过排查发现,指定的nvidia通道缺少Windows版本的CUDA 11.1包。最终通过更改安装命令成功解决了问题。
1412

被折叠的 条评论
为什么被折叠?



