当前位置: 首页 > news >正文

【安装GPU版本pytorch,torch.cuda.is_available()仍然返回False问题】

@TOC

第一步

检查cuda是否安装,CUDA环境变量是否正确设置,比如linux需要设置在PATH,window下环境变量编辑看看,是否有CUDA

第二步,核查python中torch版本

首先查看你环境里的pytorch是否是cuda版本,我这里是conda安装的,使用conda list查看是这样:
在这里插入图片描述
圈出来带cuda,那安装的肯定时cuda版本,但是torch.cuda.is_available()还是False,这种情况就比较费解,重点来了,我们去核对下python环境中用的是什么

在这里插入图片描述
可以很清楚的看到这里用的是cpu版本
所以环境使用的确实是cpu才导致加载不到gpu版本,可以卸载torch,重新安装一下

conda uninstall pytorch

然后从官网重新下载一下 ,记住这里有个坑,如果你配置了多个镜像源,这个镜像源如果没有gpu版本,可能还是会下载cpu版本
下载下来以后一定要核对一下

第三步,核查显卡驱动和cuda版本是否适配

如果上述没有问题,那么检查驱动和cuda的版本是否适配

NVIDIA-smi

在这里插入图片描述
这两个版本是否适配需要在官网查看
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#major-components

http://www.lryc.cn/news/140643.html

相关文章:

  • Git 版本控制系统
  • 70吨服务区生活污水处理设备加工厂家电话
  • 十一、hadoop应用
  • Pytorch06-复杂模型构建
  • iPhone 15 Pro与谷歌Pixel 7 Pro:哪款相机手机更好?
  • react通过ref获取函数子组件实例方法
  • MathType7MAC中文版数学公式编辑器下载安装教程
  • python项目实战
  • 网络渗透day03-Windows Server相关知识
  • 关于述职答辩的一点思考和总结
  • 远程调试环境配置
  • C++:构造方法(函数);拷贝(复制)构造函数:浅拷贝、深拷贝;析构函数。
  • vr内容编辑软件降低了虚拟现实项目开发门槛
  • 【水平垂直居中布局】CSS实现水平垂直居中的5种方法(附源码)
  • 原生js插入HTML元素
  • 腾讯云V265/TXAV1直播场景下的编码优化和应用
  • 牛客练习赛114 G-图上异或难题(线性基)
  • Neo4j之ORDER BY基础
  • 【C++杂货铺】探索vector的底层实现
  • MybatisPlus(1)
  • 探索未来世界,解密区块链奥秘!
  • win10 下运行 npm run watch-poll问题
  • Android平台RTMP|RTSP直播播放器功能进阶探讨
  • Centos7安装Telnet服务
  • 【C++】GCC对应C++的版本支持
  • 前端面试:【算法】排序、查找、递归、动态规划
  • RK3399 开机自启一个shell脚本,一直起不来BUG
  • [MyBatis系列④]核心配置文件
  • 系统架构设计高级技能 · 层次式架构设计理论与实践
  • Nuxt3打包部署到Linux(node+pm2安装和运行步骤+nginx代理)