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

相关推荐
chenchihwen1 分钟前
用Prompt 技术【提示词】打造自己的大语言智能体
人工智能·prompt
世界的尽头在哪里5 分钟前
python测试框架之pytest
开发语言·python·测试工具·单元测试·pytest
_一条咸鱼_10 分钟前
Python 语法之注释详解(二)
人工智能·python·面试
小洛~·~21 分钟前
《深度学习入门:基于Python的理论与实现》第四章神经网络的学习—上篇(损失函数登场)
python·深度学习·神经网络
bjxiaxueliang21 分钟前
一文详解opencv-python环境搭建:Mac配置python的cv2开发环境
python·opencv·macos
用户40937927136827 分钟前
并发编程
python
Eugene__Chen40 分钟前
java IO/NIO/AIO
java·python·nio
机器之心41 分钟前
AI也要007?Letta、伯克利提出「睡眠时间计算」,推理效率翻倍还不加钱
人工智能
满怀101541 分钟前
【Python核心库实战指南】从数据处理到Web开发
开发语言·前端·python
机器之心44 分钟前
WSDM 25唯一最佳论文:从谱视角揭开推荐系统流行度偏差放大之谜
人工智能