Функция torch.set_default_device в PyTorch

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.set_default_deviceЭто функция в PyTorch, используемая для установки устройства по умолчанию.

Определение функции

torch.set_default_device(device)

Пример использования

Пример

import torch

# Установить устройство по умолчанию
if torch.cuda.is_available():
    torch.set_default_device('cuda')
    x = torch.zeros(3, 4)
    print("Устройство по умолчанию:", x.device)
    torch.set_default_device('cpu')
else:
    print("CUDA is not available")

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения