torch.accelerator.set_device_idx

torch.accelerator.set_device_idx(device, /)

將當前裝置索引設定為給定裝置。

引數

device (torch.device, str, int) – 給定裝置，必須與當前加速器裝置型別匹配。

注意

如果裝置索引為負，此函式將不做任何操作。