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

安装pytorch报错torch.cuda.is_available()=false的解决方法

参考文章:
https://blog.csdn.net/qq_46126258/article/details/112708781
https://blog.csdn.net/Andy_Luke/article/details/122503884
https://blog.csdn.net/anmin8888/article/details/127910084
https://blog.csdn.net/zcs2632008/article/details/127025294

问题描述

在安装torch之后,命令行(Anaconda Powershell Prompt)运行这三行代码:

python # python
import torch
torch.cuda.is_available()

返回结果始终为False。

问题检查

方法一:

conda list

在这里插入图片描述
如果是清华镜像加速下载的,无论你是重装多少次,都一样,torch.cuda.is_available()返回false。
因为你被conda镜像安装给坑了。你以为下载的是GPU版本,其实镜像下载的是cpu版本你必须手动去官网用pip下载安装。

解决办法

首先判断自己是否适合使用GPU版本的torch

首先判断自己是否适合使用GPU版本的torch,打开自己的cmd输入

NVIDIA-smi

在这里插入图片描述
这里的11.1 指的是可驱动的最高版本。下载CUDA的版本应低于11.1。

安装cuda

1、下载低于11.1的cuda安装包
下载地址 https://developer.nvidia.com/cuda-downloads
2、安装cuda
安装教程
https://blog.csdn.net/zcs2632008/article/details/127025294
3、安装完成后验证本地安装的cuda版本
win+R,输入cmd,输入nvcc --version查看版本号,输入set cuda查看设置的环境变量
在这里插入图片描述
在这里插入图片描述

安装cudnn

1、下载安装包
下载链接
https://developer.nvidia.com/rdp/cudnn-archive
下载cuda版本对应的cudnn
2、安装
安装教程
https://blog.csdn.net/zcs2632008/article/details/127025294
3、验证
在这里插入图片描述
在这里插入图片描述

安装gpu版本的pytorch,1.7.0

1、在该网址找到cuda对应的pytorch安装语句
https://pytorch.org/get-started/previous-versions/
在这里插入图片描述2、在使用的虚拟环境下执行安装语句
3、安装完成后执行conda list,查看该虚拟环境下的包,已经不是cpu-only了
在这里插入图片描述

4、测试,
在这里插入图片描述

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

相关文章:

  • 自定义表格的表头根据后端的数据进行筛选是否进行自定义表头添加按钮
  • 海上风电应急救援vr模拟安全培训提高企业风险防范能力
  • 自定义SpringBoot启动图标
  • diffusers-Tasks
  • 文件下载漏洞, 漏洞原理, 测试方法, 漏洞防御, 常见敏感路径
  • 【零参考GAN:Pansharpening】
  • Nacos 注册中心介绍与实操
  • 基于51单片机的智能手机充电器设计
  • nginx 和gateway配置实现动静分离和反向代理
  • 【深度学习实验】网络优化与正则化(二):基于自适应学习率的优化算法详解:Adagrad、Adadelta、RMSprop
  • 系统韧性研究(3)| 工程系统韧性要求
  • .net 5 发布后swagger页面不显示问题
  • Spring Boot 3 整合 xxl-job 实现分布式定时任务调度,结合 Docker 容器化部署(图文指南)
  • 1985-2020年我国30m土地利用覆盖数据介绍
  • Django 社区志愿者管理系统
  • wordpress如何修改数据库里用户ID下一个自增值的开始数字
  • 利用chatgpt大语言模型来做数据预处理
  • 【机器学习】五、贝叶斯分类
  • k8s 资源管理方式
  • Golang Gin 接口返回 Excel 文件
  • 实战之巧用header头
  • [AUTOSAR][诊断管理][ECU][$36] 数据传输
  • sw 怎么装新版本
  • 正点原子嵌入式linux驱动开发——Linux 音频驱动
  • conda相关的命令操作
  • 如何快速使用Vue3在electron项目开发chrome Devtools插件
  • 干洗店服务预约小程序有什么作用
  • 【跟小嘉学 Rust 编程】三十四、Rust的Web开发框架之一: Actix-Web的进阶
  • 软件安装(1)——Xshell安装
  • Kafka基本原理、生产问题总结及性能优化实践 | 京东云技术团队