栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

解决torch.cuda.is

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

解决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

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/714821.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号