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

最新推荐文章于 2024-05-17 09:03:13 发布

学习中.....臭丫头

最新推荐文章于 2024-05-17 09:03:13 发布

阅读量1.6k

点赞数 4

文章标签：机器学习 pytorch python 深度学习人工智能

本文链接：https://blog.csdn.net/qq_51225231/article/details/120928563

版权

出现这个问题是有很多种原因，比如：

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，累了累了，天天都在调错的路上...........

关注