解决torch.cuda.is_available()一直返回False,将安装gpu版本的pytorch
- conda create -n pytorch python=3.6 是创建一个虚拟环境pytorch,我的python版本是3.6的。
这就是以下原因:
2. 在c盘用户里的.condarc中按照修改该文件内容
清华源网站是:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
注意在https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/这个地址是没有gpu版本的,但是可以找到cudatoolkit
在里面搜索
所以选择合适的地址很重要,在:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/找到pytorch按照自己python版本下载
3在清华源里找到相应的包下载好之后,在anaconda prompt里面输入conda install --use-local C:UsersjackVDownloadspytorch-1.10.1-py3.6_cuda10.2_cudnn7_0.tar.bz2
虽然清华源里下载快,但是直接调用命令会下载到cpu版本的,不管是直接复制pytorch里面的命令,所以手动选取cuda版本的pytorch,然后再本地导入包,然后在python里面import torch和torch.cuda.is_available()的结果是不是True
参考:https://blog.csdn.net/dysljxjsq/article/details/104159802
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)