Vérifiez si Cuda est présent dans la torche

torch.cuda.is_available()
Sad Santa