查看GPU是否可以使用

方法一：

import torch

print(torch.__version__)  # 查看torch当前版本号

print(torch.cuda.device_count()) # 查看可用的CUDA数量

print(torch.version.cuda)  # 编译当前版本的torch使用的cuda版本号

print(torch.cuda.is_available())  # 查看GPU是否可以使用

print(torch.cuda.get_device_properties(0)) # 设备的硬件配置

方法二：

命令行运行   nvidia-smi

方法三：

命令行运行   nvcc -V

查看当前GPU版本：

lspci | grep -i nvidia

当都显示有问题解决不了的时候，也许是被禁用了，去设备管理器的显示适配器启用就可以了