PyTorch 怎么修改默认device？

Question

1. cpu训练神经网络太慢，需要使用gpu
2. PyTorch需要在每次创建 tensor/model 手动 指定/更换 到cuda
3. 有办法在导包之后指定device吗？这样，之后的 tensor/model 自动创建在gpu上

Answer 1

没有方法，
只能

device = device = torch.device('cuda',0)
# 在后续的 torch.Tensor 和 torch.nn.Module 后 添加 to(device)