码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2019-11-09 19:32:22      阅读:113      评论:0      收藏:0      [点我收藏+]

标签:查看   win   --   命令行   driver   super   force   环境   version   

为什么 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.bz2conda install --use-local cudatoolkit-9.2-0.tar.bz2,即可使用下载的包进行安装。


我的大本营

学会这三个排版原则,你就能超过80%人的排版

你想过将你的人生游戏化吗?

用游戏的思路激励自己

寻找有趣或更有效率的事、工具和教程

技术图片

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

标签:查看   win   --   命令行   driver   super   force   环境   version   

原文地址:https://www.cnblogs.com/zhouzhiyao/p/11827267.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!