Definitely a regression then! Since we don’t have access to your code, could you try to come up with a small reproducing example? If this commit 2 doesn’t OOM, you can use torch.cuda.max_memory_allocated() to print out the memory usage (http://pytorch.org/docs/master/cuda.html#memory-management 8) for the small example.