Pytorch测试开发环境语句

验证GPU环境

1.引入库

代码如下（示例）：

import torch

2.输出pytorch版本

print(torch.__version__)

3.输出CUDA版本

print(torch.version.cuda)

4.输出cuDNN版本

print(torch.backends.cudnn.version())

5.输出显卡名称

print(torch.cuda.get_device_name(0))

6.测试GPU是否生效

print(torch.cuda.is_available())

---