PyTorch 怎么修改默认device？

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

没有方法，
只能

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

#指定第一块gpu
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

有帮助望采纳！谢谢