Fonctions torch.cuda de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.cudaIl s'agit d'un module de PyTorch pour la gestion CUDA. Il fournit des fonctions telles que la gestion des périphériques CUDA, la gestion de la mémoire et la synchronisation.

Définition de fonction

torch.cuda

Le module torch.cuda contient les fonctions couramment utilisées suivantes :

  • torch.cuda.is_available()- Vérifier si CUDA est disponible
  • torch.cuda.device_count()- Obtenir le nombre de périphériques CUDA
  • torch.cuda.current_device()- Obtenir le périphérique actuel
  • torch.cuda.synchronize()- Synchroniser les opérations CUDA
  • torch.cuda.empty_cache()- Vider le cache

Exemple d'utilisation

Exemple

import torch

# Vérifier si CUDA est disponible
print(f"CUDA disponible: {torch.cuda.is_available()}")

# Obtenir le nombre de périphériques CUDA
if torch.cuda.is_available():
    print(f"Nombre de périphériques CUDA : {torch.cuda.device_count()}")
    print(f"Nom du périphérique CUDA: {torch.cuda.get_device_name(0)}")

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions