【安装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

相关推荐
智行众维几秒前
自动驾驶的“虚拟驾校”如何炼成?
人工智能·自动驾驶·汽车·智能驾驶·智能网联汽车·智能驾驶仿真测试·智驾系统
空白到白33 分钟前
NLP-注意力机制
人工智能·自然语言处理
fen_fen36 分钟前
Java打包时,不将本地Jar打包到项目的最终 JAR 中
开发语言·python·pycharm
大千AI助手2 小时前
指数分布:从理论到机器学习应用
人工智能·机器学习·参数估计·概率密度函数·mle·指数分布·累积分布函数
MATLAB代码顾问2 小时前
MATLAB绘制多种混沌系统
人工智能·算法·matlab
搬砖的小码农_Sky2 小时前
人形机器人:Tesla Optimus的AI集成细节
人工智能·ai·机器人
做运维的阿瑞2 小时前
2025 年度国产大模型「开源 vs. 闭源」深度评测与实战指南
人工智能·低代码·开源
渡我白衣3 小时前
深度学习入门(三)——优化算法与实战技巧
人工智能·深度学习
可触的未来,发芽的智生3 小时前
触摸未来2025.10.10:记忆的种子,当神经网络拥有了临时工作区,小名喜忆记系统
人工智能·python·神经网络·机器学习·架构
极客BIM工作室3 小时前
演化搜索与群集智能:五种经典算法探秘
人工智能·算法·机器学习