torch.accelerator.set_device_idx#
- torch.accelerator.set_device_idx(device, /)[源代码]#
(已弃用) 将当前设备索引设置为给定的设备。
- 参数
device (
torch.device, str, int) – 给定的设备，必须与当前的 accelerator 设备类型匹配。
警告
torch.accelerator.set_device_idx()已弃用，推荐使用torch.accelerator.set_device_index()，将在未来的 PyTorch 版本中移除。
