PyTorch——device与cuda.device用法

1 查看当前的device

• 输入情况：

import torch
print("Default Device : {}".format(torch.Tensor([4, 5, 6]).device))

• 输出情况：

Default Device : cpu

2 cpu设备可以使用“cpu:0”来指定

• 输入情况

device = torch.Tensor([1, 2, 3], device="cpu:0")