torch.xpu.set_device

torch.xpu.set_device(device) [исходный код]

Установить текущее устройство.

Параметры:

device (torch.device or int or str) – выбранное устройство. Эта функция ничего не делает, если значение этого аргумента отрицательное.