解决:torch.cuda.is_available() is False

出现这个问题是有很多种原因,比如:

1.系统驱动版本低,需要更新

2.cuda版本高,驱动不支持,以及没配置好驱动支持的cuda版本

3.cuda和torch版本不匹配,

4.安装的torch的指令不对,导致没有安装支持cuda的torch

我出错的原因是版本不一致,最终还是把各个版本匹配出来了

我使用的各个版本  python3.8 + cuda 10.2版本

然后在终端执行命令:安装这些版本的包

pip install torch==1.8.1+cu102 torchvision==0.9.1+cu102 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

等待安装完成,进入下面这个:

 

再去依次执行:

activate  你的环境名

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

 最后打印出来为true,累了累了,天天都在调错的路上...........

 

你可能感兴趣的:(机器学习,pytorch,python,深度学习,人工智能)