CellProfiler / CellProfiler-plugins

Community-contributed and experimental CellProfiler modules.
http://plugins.cellprofiler.org/
56 stars 67 forks source link

stardist plugin does not free gpu memory #196

Open sunyi000 opened 1 year ago

sunyi000 commented 1 year ago

we installed both runcellpose and runstardist plugins to cellprofiler 4.2.5 no problem running either one alone on GPU.

If we create a pipeline with runstardist module before runcellpose, we have GPU out of memory issue and as a result, the next cellpose module will run on CPU instead.

Our GPU memory is approx. 11G. It seems that the startdist module did not free gpu memory after it finishes. I'm not sure if this is to do with how cellprofiler calls the run() method in runstardist.py.

could someone please help? Thank you