Verifique se Cuda está presente na tocha

torch.cuda.is_available()
Sad Santa