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.

Hi, Could you share which PyTorch package do you use? Do you install it from the link shared in this topic? [image] PyTorch for Jetson Jetson Nano Below are pre-built PyTorch pip wheel installers for Jetson Nano, TX1/TX2, Xavier, and Orin with JetPack 4.2 and newer. D…

Hi AastaLLL, Sorry, I forgot to mention my pytorch version. My pytorch version is 1.6 (torch-1.6.0-cp36-cp36m-linux_aarch64.whl ), and it was installed from Jetson Zoo.

Pytorch release memory problem

Robotics & Edge Computing Jetson Systems Jetson TX2

AastaLLL January 7, 2021, 5:42am 3

Hi,

Could you share which PyTorch package do you use?
Do you install it from the link shared in this topic?

Thanks.

Topic		Replies	Views
JetPack 4.6.1 (L4T R32.7.1): PyTorch allocates all the memory + swap! Jetson Nano cuda , pytorch	4	663	November 22, 2023
Pytorch in cuda mode use huge memory Jetson Nano pytorch	2	1422	June 28, 2021
Pytorch on jetson tx1 Jetson TX1	4	1606	October 6, 2018
How can I release GPU memory without terminating the execution process TensorRT tensorrt , python	2	1900	June 10, 2022
GPU memory cannot be released Deep Learning (Training & Inference)	0	1427	October 26, 2018
how to use DataParallel on TX2? Jetson TX2	7	730	January 15, 2019
GPU Bar1 Memory Usage couldn't release after running Pytorch code Frameworks (archived) cuda , pytorch	1	691	October 31, 2023
Pytorch 2.4.1 compilation on jetson xavier nx Jetson AGX Xavier cuda , nvbugs , pytorch	11	183	March 10, 2026
Pytorch with jetpack 4.2 works slowly than 3.3 Jetson TX2	5	1485	April 26, 2019
Does TX2 support to Pytorch 1.0 ? Jetson TX2	1	449	March 29, 2019

Pytorch release memory problem

Related topics