PyTorch torch.cuda Funciones
Pytorch torch Manual de referencia
torch.cudaEs un módulo en PyTorch para la gestión de CUDA. Proporciona funciones como gestión de dispositivos CUDA, gestión de memoria, sincronización, etc.
Definición de funciones
torch.cuda
El módulo torch.cuda incluye las siguientes funciones comunes:
torch.cuda.is_available()- Verificar si CUDA está disponibletorch.cuda.device_count()- Obtener el número de dispositivos CUDAtorch.cuda.current_device()- Obtener el dispositivo actualtorch.cuda.synchronize()- Sincronizar operaciones CUDAtorch.cuda.empty_cache()- Limpiar la caché
Ejemplo de uso
Ejemplo
import torch
# Verificar si CUDA está disponible
print(f"CUDA available: {torch.cuda.is_available()}")
# Obtener el número de dispositivos CUDA
if torch.cuda.is_available():
print(f"CUDA device count: {torch.cuda.device_count()}")
print(f"CUDA device name: {torch.cuda.get_device_name(0)}")
# Verificar si CUDA está disponible
print(f"CUDA available: {torch.cuda.is_available()}")
# Obtener el número de dispositivos CUDA
if torch.cuda.is_available():
print(f"CUDA device count: {torch.cuda.device_count()}")
print(f"CUDA device name: {torch.cuda.get_device_name(0)}")
Otras extensiones