torch.cuda.is_availabel()=False

qwe969 · 发表于 2024-1-4 15:35:03

Ubuntu系统已安装CUDA的情况下

print(torch.cuda.is_availabel())输出是False

解决方案：
查看pytorch版本信息以及支持的CUDA最低库版本
CLI下：
python 回车
import torch 回车
torch.version 回车
会输出一行路径，复制一下
vim 路径显示如下

可以看到cuda要求最低12.1，我安装的是11.8所以会出问题
卸载重装cuda
CLI下：
nvidia-uninstall 卸载下一步一下不即可
Nvidia 官网下载run格式的文件后
sudo bash run文件路径安装后即可

繁花似锦 · 发表于 2024-10-10 09:24:42

感谢楼主分享，楼主辛苦了！

风铃的低语 · 发表于 2024-10-10 10:04:03

顶顶顶顶顶顶顶

蔚蓝星辰 · 发表于 2024-10-10 10:19:42

谢谢分享

竹影摇曳 · 发表于 2024-10-10 11:32:50

1111111111111111111

一抹微笑 · 发表于 2024-10-10 11:35:25

顶顶顶顶顶顶顶

海豚的微笑 · 发表于 2024-10-10 12:16:24

66666666666666666

梦幻泡泡﫧 · 发表于 2024-10-11 03:40:21

感谢楼主分享，楼主辛苦了！

琉璃梦境 · 发表于 2024-10-11 07:16:52

感谢楼主分享，有你更精彩！！！

笑看风云️ · 发表于 2024-10-21 10:29:06

谢谢分享