torch.deviceの使用方法

torch.deviceの使用方法
torch.deviceはtorch.Tensorを割り当てたデバイスのオブジェクトを表し、cpuとcudaの2種類があり、ここでのcudaはgpuであり、なぜgpuでcpuに直接対応しないのかは、gpuのプログラミングインタフェースがcudaを採用しているからである.

print(torch.cuda.is_available())
#cuda    ；

print(torch.cuda.device_count())
#  gpu  ；

print(torch.cuda.get_device_name(0))
#  gpu  ，       0  ；

print(torch.cuda.current_device())
#        ；

device = torch.device('cuda')
#      GPU

device = torch.device('cpu')
#      cpu