Функция PyTorch torch.get_device_module

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

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

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

torch.get_device_module(device='cuda')

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

Пример

import torch

# Получить модуль устройства CUDA
cuda_module = torch.get_device_module('cuda')
print("Модуль CUDA:", cuda_module)

# Проверить, доступна ли CUDA
print("CUDA доступна:", torch.cuda.is_available())

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

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