今天给新电脑装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