Windows系统下,电脑有nvidia的GPU,但是torch.cuda.device_count()显示GPU数量为0

为了配合学习李沐的《动手学深度学习 PyTorch版》,近期在Win 10系统的笔记本电脑中完成了相关配置。笔记本电脑配有nvidia 3060的显卡。

学习到第5章的GPU计算时,尝试运行torch.cuda.device_count(),返回GPU数目为0。为了解决这个问题,依次进行了以下操作。

1. 检查问题

出现以上问题,首先想到的是torch安装错了版本。因此进行了如下操作:

(1) 在终端输入python,进入python命令式编程环境

在这里插入图片描述

(2) 使用torch.__version__查看torch版本,发现确实是CPU版

在这里插入图片描述

2. 解决问题

(1) 使用pip uninstall torch,卸载已安装的torch

在这里插入图片描述

(2) 使用pip uninstall torchvision,卸载已安装的torchvision

在这里插入图片描述

(3) 打开Pytorch官网:https://pytorch.org/

(4) 下拉官网,找到“Previous versions of PTorch”

在这里插入图片描述

(5) 按照指令安装

Previous versions of PyTorch的网址如下:
https://pytorch.org/get-started/previous-versions/
根据系统和按照方式,找到对应版本
我是Windows系统,使用pip按照,并且根据李沐的《动手学深度学习 PyTorch版》中文版,安装pytorch 1.12.0
在这里插入图片描述
找到对应的按照命令,复制到终端,执行安装
安装成功截图如下:
在这里插入图片描述

3. 验证问题是否解决

(1) 在终端输入python,进入python命令式编程环境

在这里插入图片描述

(2) 导入torch包,重新查看GPU数目

import torch
torch.cuda.device_count()
在这里插入图片描述
显示1,问题解决

  • 6
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值