1. torch.device代表的含义

torch.device代表将torch.tensor分配到的设备的对象(简单点说，就是分配到你的CPU还是GPU上,以及哪块GPU上)。

2. torch.device的参数解读

torch.device包含一个设备类型（‘cpu’或者‘cuda’设备类型）和可选的设备的序号。如果设备序号不存在，则为当前设备；
torch.tensor使用参数cuda和使用cuda:X的效果是一样的，其中X是torch.cuda.current_device()对应的结果。

3. 使用torch.device的方式

1. 通过字符串

举例如下：

>>> torch.device('cuda:0')
device(type='cuda', index=0)

>>> torch.device('cpu')
device(type='cpu')

>>> torch.device('cuda')  # current cuda device
device(type='cuda')

2. 通过字符串和设备编号的形式

举例如下：

>>> torch.device('cuda', 0)
device(type='cuda', index=0)

>>> torch.device('cpu', 0)
device(type='cpu', index=0)

4. 资料参考

https://ptorch.com/news/187.html