search

nadermx / backgroundremover

Background Remover lets you Remove Background from images and video using AI with a simple command line interface that is free and open source.
https://www.backgroundremoverai.com
MIT License
6.69k stars 556 forks source link

Proper gpu memory release #140

Closed Isalia20 closed 5 months ago

Isalia20 commented 5 months ago

Without calling cuda empty cache, memory on GPU isn't released(even if function ends, it keeps the memory occupied before the process ends). This line helps GPU release the memory

nadermx commented 4 months ago

I must thank personally thank you for this, kept having ghost GPU memory issues. This seems to have fixed it.

Isalia20 commented 4 months ago

No problem :) Glad it helped