device

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

Менеджер контекста, который меняет выбранное устройство.

Параметры:

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