【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