torch.device使用方法

torch.device代表将torch.Tensor分配到的设备的对象，有cpu和cuda两种，这里的cuda就是gpu，至于为什么不直接用gpu与cpu对应，是因为gpu的编程接口采用的是cuda

import torch
#pytorch的版本
torch.__version__
#是否支持CUDA
torch.cuda.is_available()
#CUDA的版本
print(torch.version.cuda)
#cuDNN的版本
print(torch.backends.cudnn.version())
# 查看当前默认的CUDA设备
print(torch.cuda.current_device())
# 查看CUDA = 0设备的属性
print(torch.cuda.get_device_properties(device=0))
# 查看当前CUDA设备的总内存和已使用内存
print(torch.cuda.memory_summary(device='cuda'))
#GPU算力
torch.cuda.get_device_capability(device=0)