欢迎光临散文网 会员登陆 & 注册

torch.cuda.is_available()返回false

2022-02-20 00:15 作者:wing2791  | 我要投稿

本人误打误撞似乎解决了这个问题,记录一下,如果同学没有其他的解决方案,可以参考本文。

在一篇博客上看到,说问题是清华源有误,下载下来的就是cpu版本,本人也尝试了很多方法,最后使用离线下载解决,简单写一下步骤:


步骤一:下载

下载链接:可用迅雷下载

https://download.pytorch.org/whl/torch_stable.html  

cu102:   cuda版本10.2

torch-1.7.0:   torch版本1.7.0

cp37:    python版本3.7(后面的cp37m不知道是啥)

linux:    操作系统linux

x86_64:  同时兼容32和64位系统

下载示例


步骤二:安装

将上面的包放入下面的位置(其他位置不知道行不行),在上面的地址栏输入cmd,输入命令,中间第三个参数是自己下载的包名,后面的参数是豆瓣源,不知道有没有用,都试试

pip install "torch-1.6.0-cp36-cp36m-win_amd64.whl" -i https://pypi.douban.com/simple

pip install "torchvision-0.7.0+cu101-cp36-cp36m-win_amd64.whl" -i https://pypi.douban.com/simple 


安装示例


步骤三:验证

测试示例

注:cuda10.2 和 cudnn安装包下载(可忽略)

链接:

https://blog.csdn.net/weixin_55775980/article/details/119768352

torch.cuda.is_available()返回false的评论 (共 条)

分享到微博请遵守国家法律