pytorch gpu不可用 原因核查

首先检查pytorch cuda 显卡驱动 之间的对应关系

查看pytorch对应的cuda版本

打开终端输入：python

输入命令：import torch

再输入：print(torch.version.cuda)


查看pytorch 版本


print(torch.__version__) 

查看 cuda 是否安装成功

显示 torch 安装成功，但 pytorch 对应的cuda版本未安装成功

查看cuda 版本

cat /usr/local/cuda/version.txt

查看是否能使用GPU

import torch
print(torch.cuda.is_available())

torch.cuda.is_available()
cuda是否可用；

torch.cuda.device_count()
返