1，编写代码

import torch

# 查看torch版本
print('--- torch版本 ---')
print(torch.__version__)

# 查看cuda版本
print('--- cuda版本 ---')
print(torch.version.cuda)

# GPU是否可用
print('--- GPU是否可用 ---')
print(torch.cuda.is_available())

# 返回gpu数量
print('--- GPU数量 ---')
print(torch.cuda.device_count())

# 返回gpu名字，设备索引默认从0开始
print('--- GPU名称 ---')
n = 0
while n < torch.cuda.device_count():
    print(torch.cuda.get_device_name(n))
    n += 1

# 代码测试
print('--- PyTorch代码测试 ---')
print(torch.rand(3,3))
print('--- PyTorch代码测试（在GPU上测试PyTorch代码） ---')
print(torch.rand(3,3).cuda())

2，运行验证代码