【FAQ】P3.为什么torch.cuda.is_available()是False
- 1、下载文档前请自行甄别文档内容的完整性,平台不提供额外的编辑、内容补充、找答案等附加服务。
- 2、"仅部分预览"的文档,不可在线预览部分如存在完整性等问题,可反馈申请退款(可完整预览的文档不适用该条件!)。
- 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,即可使⽤下载的包进⾏安装。
我的⼤本营
寻找有趣或更有效率的事、⼯具和教程。