lbd-hfut / Computing-Platform

Solving digital image correlation based on physical information network
MIT License
13 stars 0 forks source link

Memory leak on train step #2

Open SSDGADsss opened 1 week ago

SSDGADsss commented 1 week ago

When I train my image college, I noticed that the memory usage of the graphics card increases as the number of images increases, maybe this is a bug.

I train 160 images use my 4070 Super GPU, Occupying over 12GB of video memory. I use nvtop command to monitor GPU active, see below:

截图 2024-10-11 18-21-48

Yellow line is gpu memory use, when train a new image , the last train's GPU memoy is not release.