验证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())