PyTorch torch.get_device_module Funktion

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

torch.get_device_moduleEs ist eine Funktion in PyTorch, mit der das aktuelle Gerätemodul abgerufen wird.

Funktionsdefinition

torch.get_device_module(device='cuda')

Anwendungsbeispiel

Beispiel

import torch

# CUDA-Gerätemodul abrufen
cuda_module = torch.get_device_module('cuda')
print("CUDA-Modul:", cuda_module)

# Prüfen, ob CUDA verfügbar ist
print("CUDA verfügbar:", torch.cuda.is_available())

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

Andere Erweiterungen