torch.cuda.is_available(): Verificarea Disponibilității Acceleratorului GPU în PyTorch & Deep Learning Practic

Funcția torch.cuda.is_available() returnează True dacă binarul PyTorch a fost compilat cu suport CUDA, driverul NVIDIA este încărcat în sistem și există cel puțin un GPU compatibil disponibil pentru calcule numerice.

Șablonul standard de selecție dispozitiv

import torch

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print("Dispozitiv selectat:", device)

if torch.cuda.is_available():
    print("Număr GPU-uri:", torch.cuda.device_count())
    print("Nume GPU:", torch.cuda.get_device_name(0))

Este primul test executat în orice script de antrenare sau inferență pentru a preveni căderile pe sisteme fără accelerator grafic.