Функция torch.device в PyTorch
Справочное руководство PyTorch torch
torch.deviceявляется функцией в PyTorch, используемой для создания объектов устройства и указания устройства, на котором хранятся тензоры.
Определение функции
torch.device(device)
Примеры использования
Примеры
import torch
# Создать устройство CPU
cpu_device = torch.device('cpu')
print("Устройство CPU:", cpu_device)
# Создать устройство CUDA
if torch.cuda.is_available():
cuda_device = torch.device('cuda:0')
print("Устройство CUDA:", cuda_device)
# Создать тензор на GPU
x = torch.zeros(3, 4, device=cuda_device)
print("Устройство тензора:", x.device)
# Создать устройство CPU
cpu_device = torch.device('cpu')
print("Устройство CPU:", cpu_device)
# Создать устройство CUDA
if torch.cuda.is_available():
cuda_device = torch.device('cuda:0')
print("Устройство CUDA:", cuda_device)
# Создать тензор на GPU
x = torch.zeros(3, 4, device=cuda_device)
print("Устройство тензора:", x.device)
Другие расширения