为了配合学习李沐的《动手学深度学习 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,问题解决