torch.accelerator.set_device_index

torch.accelerator.set_device_index(device, /)

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

引數

device (torch.device, str, int) – 一個給定的裝置，它必須與當前的 accelerator 裝置型別匹配。

注意

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