安装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、测试,
在这里插入图片描述

  • 0
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值