参考文章
文章参考:链接.
问题
- 在pytorch官网按照指示安装 cuda版11.6的却安装成cpu版的导致没法使用cpu
- 下载11.3却下载不动,切换镜像也下载不动
解决办法
-
对于第一个问题,首先看看是否是cpu版的 使用cuda list 查看 pytorch 描述是否是cpu若是的话恭喜下载的为cpu版的,若不是的话,可能是显卡驱动 pytorch python 和cudnn版本的问题
-
对于第二个问题,先切换成清华的源,然后巨坑的一点
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
中要把 -c去掉才能在清华的源下载,11.6版的清华的源没有所以下载不了