【pytorch学习】torch.device

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('cpu')
device(type='cpu')

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

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

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

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


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