torch.set_default.device()

方法： 

torch.set_default.device()

功能： 设置默认计算张量的设备(CUDA 或者 CPU) 

注意： 这里是全局的， 如果是临时的 使用： with torch.device(device)

例子：

来自官方：torch.set_default_device — PyTorch 2.2 documentation

>>> torch.tensor([1.2, 3]).device
device(type='cpu')
>>> torch.set_default_device('cuda')  # current device is 0
>>> torch.tensor([1.2, 3]).device
device(type='cuda', index=0)
>>> torch.set_default_device('cuda:1')
>>> torch.tensor([1.2, 3]).device
device(type='cuda', index=1)

另外有一个警告和说明：

 

翻译软件的翻译：

警告：该功能在每次调用torch API时会带来一点性能损失（不仅限于工厂函数）。如果这给您造成问题，请在https://github.com/pytorch/pytorch/issues/92701上发表评论。

注意：这不影响创建与输入共享内存的张量的函数，例如 torch.from_numpy() 和 torch.frombuffer()。