Hi! I test the code you provided. I found no torch.zero (should be torch.zeros). But I found the gpu memory was not change even for Pytorch 1.1.0. This is very strange. By the way, I run this code in a docker container.