【FAQ】P3.为什么torch.cuda.is_available()是False

合集下载
  1. 1、下载文档前请自行甄别文档内容的完整性,平台不提供额外的编辑、内容补充、找答案等附加服务。
  2. 2、"仅部分预览"的文档,不可在线预览部分如存在完整性等问题,可反馈申请退款(可完整预览的文档不适用该条件!)。
  3. 3、如文档侵犯您的权益,请联系客服反馈,我们会尽快为您处理(人工客服工作时间:9:00-18:30)。

【FAQ】P3.为什么torch.cuda.is_available()是False
为什么 torch.cuda.is_available() 是 False
torch.cuda.is_available(),这个指令的作⽤是看,你电脑的 GPU 能否被 PyTorch 调⽤。

如果返回的结果是 False,可以按照以下过程进⾏排查。

1、确认你的 GPU,是否⽀持 CUDA(是否⽀持被 PyTorch 调⽤)
⾸先,确定你的显卡型号,是否是 NVIDIA 显卡。

可以从任务管理器或者设备管理器来查看显卡的型号。

之后,去看,如果其中有你的显卡型号,则说明你的显卡是⽀持被 PyTorch 调⽤的。

(绝⼤多数的 NVIDIA 显卡都是⽀持的)
如果没有 NVIDIA 显卡的话,也没有关系。

CPU 就已经⾜够了,⽽且你会在后⾯的教程看到,对于⼩型⽹络,CPU 速度更快(窃喜)
2、打开命令⾏,输⼊nvidia-smi,查看⾃⼰的Driver Version
我们教程中安装的 PyTorch 1.3 + CUDA 9.2 版本,要求电脑的显卡驱动⼤于396.26。

像我截图中的驱动版本为430.86,⼤于396.26。

如果你的驱动版本⼩于396.26,请⽤各种驱动管理软件或者软件管家,去升级你的显卡驱动。

当然,更推荐去官⽹,下载对应的最新驱动。

3、下载最新驱动。

在选择相应的显卡型号,操作系统,其他默认。

其中的 Notebooks 是指笔记本。

之后,点击搜索,下载最新驱动后,进⾏安装即可。

4、检查驱动版本。

安装完最新的驱动后,可以再次在命令⾏窗⼝输⼊nvidia-smi,查看最新的版本是否安装成功。

5、打开 Anaconda Prompt,输⼊conda activate pytorch,再输⼊python,进⼊ python 环境。

在 python 环境中,输⼊import torch, 之后输⼊torch.cuda.is_available,查看返回的结果是否是 True。

使⽤ Conda 下载 PyTorch 速度太慢了,怎么办?
1、(⽞学办法) 早上下载安装,感觉早上的时候,下载的速度明显变快。

2、从本教程最顶端的百度云处,下载这两个⽂件。

(这两个⽂件是适⽤于 pytorch1.3 + cuda9.2 + windows)
将这两个下载好的⽂件,放在 Anaconda 安装出的 pkgs ⽂件夹下。

之后打开 Anaconda Prompt,输⼊conda activate pytorch。

之后,输⼊以下指令:conda install --use-local pytorch-1.3.0-py3.6_cuda92_cudnn7_0.tar.bz2和conda install --use-local cudatoolkit-9.2-0.tar.bz2,即可使⽤下载的包进⾏安装。

我的⼤本营
寻找有趣或更有效率的事、⼯具和教程。

相关文档
最新文档