torch.cuda.is_available()返回false

首先进行排除:

1.了解自己GPU的型号(任务管理器---性能)(设备管理器---显示适配器)

确定GPU支持cuda

在anaconda prompt中输入命令:

nvidia-smi

torch.cuda.is_available()返回false_第1张图片

是否版本过低,如果过低就去更新下载nvidia驱动程序

import torch
torch.cuda.is_available()

2.下载cuda

torch.cuda.is_available()返回false_第2张图片

选择合适版本安装

3.安装pytorch

torch.cuda.is_available()返回false_第3张图片 选择合适版本之后,复制下方的命令安装

最后再次检测,出现true即可

注意:

版本一定要对应,不然就会一直出错

你可能感兴趣的:(pytorch,pytorch)