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

相关推荐
星空椰4 分钟前
快速掌握FastAPI:高效构建Web API
python·fastapi
塔尖尖儿5 分钟前
Python中range()到底是什么演示
python
scott1985125 分钟前
opencv 畸变系数的说明
人工智能·数码相机·opencv
LS_learner8 分钟前
Transmormer从零基础到精通
人工智能
ASD123asfadxv20 分钟前
【蜂巢健康监测】基于YOLO的蜂群病虫害识别系统
人工智能·yolo·目标跟踪
说私域25 分钟前
基于AI智能名片链动2+1模式服务预约商城系统的社群运营与顾客二次消费吸引策略研究
大数据·人工智能·小程序·开源·流量运营
Ethan-D37 分钟前
#每日一题19 回溯 + 全排列思想
java·开发语言·python·算法·leetcode
丝斯20111 小时前
AI学习笔记整理(50)——大模型中的Graph RAG
人工智能·笔记·学习
Coder_Boy_1 小时前
基于SpringAI的在线考试系统-DDD业务领域模块设计思路
java·数据库·人工智能·spring boot·ddd
甜辣uu2 小时前
双算法融合,预测精准度翻倍!机器学习+深度学习驱动冬小麦生长高度与产量智能预测系统
人工智能·小麦·冬小麦·生长高度·植物生长预测·玉米·生长预测