CellProfiler/CellProfiler-plugins

stardist plugin does not free gpu memory

sunyi000 opened this issue · 0 comments

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