Hi, I have tried the follow code on PC platform, it can release memory successfully, but it doesn’t work on TX2 (Jetpack 4.4) . Is there other method to release memory with Pytorch object? import gc import torch x=torch.randn((32,3,300,300),device=torch.device(‘cuda’)) del x gc.collect() to…

Hi AastaLLL, It can release memory successfully on TX2, with both pytorch 1.6 and 1.7. Tensor shape like (32,3,300,300) is too small, can not get the obvious memory release amount. So when I increase shape to (2000,3,300,300), I see the memory release result. Thank you for help.