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 263