This happens also in the windows port of PyTorch, the only way to overcome this when using (in my case) large CNN’s is to use: torch.backends.cudnn.enabled=False
.
This happens also in the windows port of PyTorch, the only way to overcome this when using (in my case) large CNN’s is to use: torch.backends.cudnn.enabled=False
.