解决torch.cuda.is

解决torch.cuda.is,第1张

解决torch.cuda.is

解决torch.cuda.is_available()一直返回False,将安装gpu版本的pytorch

    conda create -n pytorch python=3.6 是创建一个虚拟环境pytorch,我的python版本是3.6的。
如果发现解决Anaconda出现CondaHTTPError: HTTP 000 ConNECTION FAILED for url问题

这就是以下原因:
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

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/5711466.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-17
下一篇 2022-12-17

发表评论

登录后才能评论

评论列表(0条)

保存