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

相关推荐
树獭非懒10 小时前
AI大模型小白手册|Embedding 与向量数据库
后端·python·llm
KaneLogger10 小时前
【Agent】openclaw + opencode 打造助手 安装篇
人工智能·google·程序员
知识浅谈11 小时前
一步步带你把 OpenClaw 玩宕机(附云服务器避坑部署教程)
人工智能
冬奇Lab11 小时前
OpenClaw 深度解析(四):插件 SDK 与扩展开发机制
人工智能·开源·源码阅读
IT_陈寒12 小时前
SpringBoot实战:5个让你的API性能翻倍的隐藏技巧
前端·人工智能·后端
机器之心12 小时前
让AI自我进化?斯坦福华人博士答辩视频火了,庞若鸣参与评审
人工智能·openai
iceiceiceice13 小时前
iOS PDF阅读器段评实现:如何从 PDFSelection 精准还原一个自然段
前端·人工智能·ios
唐叔在学习13 小时前
就算没有服务器,我照样能够同步数据
后端·python·程序员
AI攻城狮14 小时前
RAG Chunking 为什么这么难?5 大挑战 + 最佳实践指南
人工智能·云原生·aigc