Fonction PyTorch torch.get_device_module
Manuel de référence PyTorch torch
torch.get_device_moduleIl s'agit d'une fonction dans PyTorch utilisée pour obtenir le module de périphérique actuel.
Définition de fonction
torch.get_device_module(device='cuda')
Exemple d'utilisation
Exemple
import torch
# Obtenir le module de périphérique CUDA
cuda_module = torch.get_device_module('cuda')
print("Module CUDA :", cuda_module)
# Vérifier si CUDA est disponible
print("CUDA disponible :", torch.cuda.is_available())
# Obtenir le module de périphérique CUDA
cuda_module = torch.get_device_module('cuda')
print("Module CUDA :", cuda_module)
# Vérifier si CUDA est disponible
print("CUDA disponible :", torch.cuda.is_available())
Autres extensions