这两天正在用服务器跑程序，但是发现运行速度极慢，查看正在运行的进程，也没看到自己的进程，但是程序又确确实实在运行，这就奇了怪了，一通查找，发现程序竟然是在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
pip install 下载的文件所在目录/文件名
可以检验一下是否安装成功
进入python环境，输入
import torch
print(torch.cuda.is_available())
如果输出是true，就代表安装成功了
Original: https://blog.csdn.net/didadifish/article/details/127487635
Author: didadifish
Title: cuda11.2安装pytorch——torch.cuda.is_available()=false
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/707018/
转载文章受原作者版权保护。转载请注明原作者出处！