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

相关推荐
小炫y4 小时前
bilibil
python
BizViewStudio4 小时前
甄选 2026:AI 重构新媒体代运营行业的三大核心变革与落地路径——附10家优质服务商
大数据·网络·人工智能·媒体
小兵张健4 小时前
武汉 AI 面试圈,真的很小
人工智能·面试·ai编程
z4424753264 小时前
HTML函数开发用旋转屏有优势吗_特殊硬件形态适配说明【方法】
jvm·数据库·python
阿乔外贸日记4 小时前
土耳其包装市场需求缺口分析
大数据·人工智能·物联网·搜索引擎·云计算
mahtengdbb14 小时前
RepViT块改进YOLOv26重参数化与双路径特征混合双重突破
人工智能·yolo·目标跟踪
wusp19944 小时前
DeepSeek V4 Flash + DMXAPI:文生图提示词智能优化
大数据·人工智能·机器学习
领航猿1号4 小时前
AI Coding 安全解决方案
网络·人工智能·安全
Ares-Wang5 小时前
Flask》》flask-login
后端·python·flask
李昊哲小课5 小时前
AGENTS.md - LLM Wiki 行为规范
人工智能·本地知识库·llm wiki