torch.cuda.set_device¶ torch.cuda.set_device(device)[源代码]¶ 设置当前设备。 不建议使用此函数,建议使用 device。在大多数情况下,最好使用 CUDA_VISIBLE_DEVICES 环境变量。 参数 device (torch.device 或 int) – 选定的设备。如果此参数为负数,则此函数为无操作。