在安装了CUDA的前提下,又在服务器上安装了无数遍pytorch,依然不能使用,运行代码就报错,提示CUDA不能用!
在网上找了很久,最后在大神的帖子下找到了答案。链接: 参考1 ;参考2; 参考3
一直报错的原因在于,我们在服务器上安装好了anaconda后,添加了清华源镜像,以便加速下载相关包。下载的通道也会优先使用清华源的通道,当我每次安装cuda版本的时候,这个下载通道他找不到对应的版本,就会默认下载cpu版本的pytorch,这就导致了,我不管下载多少次,下载的版本都不对!自己可以查看安装的pytorch版本,看是不是这个问题。conda list
我的如下,版本明显是cpu。
找到了问题后,我新建了一个虚拟环境,把通道改成了默认的通道。
conda config --remove-key channels
因为换了通道,再用conda下载的话,就会容易中断,所以我使用了pip下载。
这样后,问题得到解决,感谢其他大神写的帖子对我的帮助!