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

相关推荐
William_Edmund12 分钟前
Python 语言学习——应用1.2 数字图像处理(第二节,变换)
人工智能·学习·计算机视觉
子墨77728 分钟前
Flask之Hello world 详解
python
繁依Fanyi30 分钟前
旅游心动盲盒:开启个性化旅行新体验
java·服务器·python·算法·eclipse·tomcat·旅游
计算机编程-吉哥36 分钟前
计算机毕业设计 基于Python的个性化旅游线路推荐系统的设计与实现 Python+Django+Vue 前后端分离 附源码 讲解 文档
python·django·毕业设计·课程设计·毕业论文·计算机毕业设计选题·个性化旅游线路推荐系统
罔闻_spider41 分钟前
爬虫prc技术----小红书爬取解决xs
爬虫·python·算法·机器学习·自然语言处理·中文分词
我爱学Python!41 分钟前
面试问我LLM中的RAG,秒过!!!
人工智能·面试·llm·prompt·ai大模型·rag·大模型应用
python机器学习ML1 小时前
机器学习K近邻算法——python详细代码解析(sklearn)(1)
python·机器学习·近邻算法·knn
zh路西法1 小时前
基于opencv-C++dnn模块推理的yolov5 onnx模型
c++·图像处理·pytorch·opencv·yolo·dnn·yolov5
weixin_514548891 小时前
机器学习课程学习周报十五
人工智能·学习·机器学习
慢成长1 小时前
如何创建虚拟环境并实现目标检测及验证能否GPU加速
人工智能