torch.mtia.set_device

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

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

Параметры:

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