cuda11.2安装pytorch——torch.cuda.is_available()=false

这两天正在用服务器跑程序,但是发现运行速度极慢,查看正在运行的进程,也没看到自己的进程,但是程序又确确实实在运行,这就奇了怪了,一通查找,发现程序竟然是在CPU上运行,也就是

torch.cuda.is_available()=false

1、重新卸载安装
但是import torch明明是没问题的,没办法,只能卸载又重装,期间都是用官网的安装命令(参考https://blog.csdn.net/didadifish/article/details/115076778),结果还是没用
百度了各种解决方法之后,终于找到了原因
在这里插入图片描述
用官网的命令下载的竟然是CPU版本的!!!
2、离线下载
那好吧,那我先去网站下载适配我cuda=11.2的torch(https://download.pytorch.org/whl/torch_stable.html)
在这里插入图片描述

cu111:表示cuda版本为11.1,
torch-1.8.1:表示torch版本为1.8.1
cp37:表示适用python版本为3.7
linux:表示适用于linux系统
x86_64:表示同时兼容32和64位系统
大家注意:如果还要下载torchvision和torchaudio,记得选择适配的版本
在这里插入图片描述
离谱的是,网站上竟然没有对应cuda=11.2的torch版本!
难道只能重装cuda才能解决我的问题吗?
3、最终解决方法
在百度了cuda的安装方法后,深知这是一个巨大的工程,并且容易影响服务器上正在跑的别人的程序。
考虑到版本向下兼容,可能不一定非要下载cuda=11.2对应的那个版本的torch,或许低于这个版本就可以。
**注意:**由于服务器上的gpu是RTX 3090的,它适配的cuda不能低于11.1,因此还得满足这个条件
也就是11.1<=torch.cuda<=11.2
(大家可以通过nvcc -V查看自己服务器的cuda版本)
所以最终选择了下载cuda=11.1对应的torch
在这里插入图片描述
然后安装

pip install 下载的文件所在目录/文件名

可以检验一下是否安装成功
进入python环境,输入

import torch
print(torch.cuda.is_available())

如果输出是true,就代表安装成功了

  • 7
    点赞
  • 50
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值