完美解决torch.cuda.is_available()一直返回False的玄学方法
正好趁此假期,疫情原因无法出去浪,在家学一下pytorch。但是我按照网上的所有方法,torch.cuda.is_available()总是返回False。安装正确在cmd中敲nvcc -V能显示即可。conda+重新开放的清华源=神速。但我自己的情况并不符合。偶然发现,使用官网conda命令下载的,居然全是CPU版本。无论我怎么尝试使用conda下载任何版本,都是cpu版本的。在此处寻找符合自己版本的pytorch-cuda版本,torchvision版本下载。但是不会断了。。。。勉强能下)下载后pip install xxx 安装即可。但是奇迹是怎么发生的,conda为什么只能下cpu版本,我一直没有搞明白。目前的解决方法有两个,可以去清华源网站上找对应的版本,
用户评论