Sprawdź, czy CUDA jest obecna w pochodniach

torch.cuda.is_available()
Sad Santa