THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp

By the way, for anyone following this topic, a solution to at least get things running is to set `torch.backends.cudnn.benchmark = False’. I found this solution in this thread: A error when using GPU