torch.cuda.set_device¶ torch.cuda.set_device(device)[源代码]¶ 设置当前设备。 不鼓励使用此函数，建议使用 device。在大多数情况下，使用 CUDA_VISIBLE_DEVICES 环境变量会更好。 Parameters 设备 (torch.device 或 整数) – 选择的设备。如果此参数为负数，则此函数不执行任何操作。
