search

MouseLand / cellpose

a generalist algorithm for cellular segmentation with human-in-the-loop capabilities
https://www.cellpose.org/
BSD 3-Clause "New" or "Revised" License
1.4k stars 403 forks source link

[Doc] Describe how to manage multithreading #1045

Open royjacobson opened 2 weeks ago

royjacobson commented 2 weeks ago

Spent a bit of time on figuring this out (after accidentally running on all 256 cores of our shared login server 😅) so figured it is worth a bit of documenting. Please LMK if there's a more appropriate place for this.

ashishUthama commented 2 weeks ago

Might also be worth pointing to https://pytorch.org/docs/stable/generated/torch.set_num_threads.html (Since that ought to keep up with whatever library PyTorch uses under the hood)