【安装GPU版本pytorch,torch.cuda.is_available()仍然返回False问题】

@TOC

第一步

检查cuda是否安装,CUDA环境变量是否正确设置,比如linux需要设置在PATH,window下环境变量编辑看看,是否有CUDA

第二步,核查python中torch版本

首先查看你环境里的pytorch是否是cuda版本,我这里是conda安装的,使用conda list查看是这样:

圈出来的torch带cuda,那安装的肯定时cuda版本,但是torch.cuda.is_available()还是False,这种情况就比较费解,重点来了,我们去核对下python环境中用的是什么

可以很清楚的看到这里用的是cpu版本,也就是说环境中确实存在两个版本

所以环境使用的是cpu才导致加载不到gpu版本,可以卸载torch,重新安装一下

bash 复制代码
conda uninstall pytorch

然后从官网重新下载一下 ,记住这里有个坑,如果你配置了多个镜像源,这个镜像源如果没有gpu版本,可能还是会下载cpu版本

下载下来以后一定要再次核对一下,或者下载的时候指定下载的镜像源

第三步,核查显卡驱动和cuda版本是否适配

如果上述没有问题,那么检查驱动和cuda的版本是否适配

bash 复制代码
NVIDIA-smi

这两个版本是否适配需要在官网查看

https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#major-components

相关推荐
ws201907几秒前
产业链协同创新,2026广州汽车零部件展赋能智造未来
人工智能·科技·汽车
HIT_Weston1 分钟前
174、【Agent】【OpenCode】TuiThreadCmd(类型补丁)
人工智能·agent·opencode
星火10243 分钟前
【LangChain4j系列06】ChatMemory 对话记忆管理
人工智能·后端
数据百晓通5 分钟前
AI 重构治理逻辑:2026 数据治理厂商的能力进化与分水岭
人工智能·重构
跨境小彭5 分钟前
Temu运营优化方案:活动库存自动化管控,解决断流、超卖与权重下跌问题
大数据·运维·人工智能·自动化·跨境电商·temu·temu电商运营
烟雨江南7858 分钟前
医院门诊医患沟通如何实时转写?——灵声智库流式 ASR、医学术语与双角色记录实践
人工智能·语音识别·agent
树码小子8 分钟前
Pycharm解释器配置问题解决
ide·python·pycharm
风途科技~8 分钟前
峰值日照时数精准测,这款光伏电站环境监测仪器助力电站发电量评估
大数据·人工智能
watersink9 分钟前
机器学习随机森林
人工智能·随机森林·机器学习
AI多Agent协作实战派13 分钟前
AI多Agent协作系统实战(三十九):AI修对了Bug,却越了权——自动化团队的第一条铁律
人工智能·自动化·bug