阐明:torch.cuda.is_available()这个指令的作用是看你电脑的 GPU 是否被 PyTorch 调用。
如果返回的后果是 False,能够依照以下过程进行排查。

Step1:确认硬件反对,确认你的 GPU是否反对 CUDA(是否反对被 PyTorch 调用)

1.确定计算机中是否是独立显卡,是否是 NVIDIA 显卡。能够从 工作管理器 或者 设施管理器 来查看显卡的型号。

  • 工作管理器->性能->GPU:

  • 我的电脑->属性->设施管理器->显示适配器:

有“NVIDIA GeForce 840M”字样信息示意有独立显卡。

2.去 NVIDIA官网 查看其中是否有你的显卡型号,如果有,则阐明你的显卡是反对被 PyTorch 调用的。

Step2:查看显卡驱动版本并更新。

1.关上命令行,输出 nvidia-smi,查看本人的 “Driver Version”

注:如呈现如下报错:

'nvidia-smi' 不是外部或外部命令,也不是可运行的程序

请参考文末“可能遇到的问题”进行解决。

一般而言,不同版本的CUDA要求不同的NVIDIA驱动版本,同时显卡驱动版本要不低于CUDA的装置版本。

CUDA对显卡驱动版本要求如图:

比方我装置的是 PyTorch 1.5 + CUDA 9.2 版本,要求电脑的显卡驱动大于398.26。

2.如果驱动版本太低,在官网,下载对应的最新驱动并装置更新。

抉择相应的显卡型号,操作系统,下载类型和语言默认。其中的 Notebooks 是指笔记本。

之后,点击搜寻,下载最新驱动后,依照指引进行装置即可。

Step3:验证驱动版本及GPU是否可调用。

1.再次在终端窗口输出 nvidia-smi,查看最新的版本是否装置胜利。
2.进入python环境:

conda activate pytorchpython

3.在python环境中:

import torchtorch.cuda.is_available()

查看返回后果是否是True

可能遇到的问题:
1.执行'nvidia-smi' 命令时提醒'nvidia-smi'不是外部或外部命令,也不是可运行的程序

起因:因为它找不到该命令。这个文件是一个exe文件,个别都在上面这个文件夹中。

C:\Program Files\NVIDIA Corporation\NVSMI

所以想要应用该命令必须要能失去这个文件夹,能力执行。

解决:增加环境变量

我的电脑->右键->属性->高级零碎设置->高级->环境变量->零碎变量->path->编辑->新建->拷贝nvidia-smi.exe文件所在门路(个别为C:\\Program Files\\NVIDIA Corporation\\NVSMI

参考:
https://www.cnblogs.com/zhouz...
https://blog.csdn.net/shuiyix...
https://docs.nvidia.com/cuda/...