Проверьте, присутствует ли CUDA в Torch

torch.cuda.is_available()
Sad Santa