search

kevinjohncutler / omnipose

Omnipose: a high-precision solution for morphology-independent cell segmentation
https://omnipose.readthedocs.io
Other
96 stars 29 forks source link

Issue with old pytorch version #17

Closed lguerard closed 1 year ago

lguerard commented 1 year ago

Hi,

I have been having issue with omnipose in the last couple of days and found out that the mps support in your latest version of Cellpose is giving issues with old version of pytorch (1.8.2) as it doesn't have backends. Would it be possible to bypass this for older pytorch version ?

Thanks !

kevinjohncutler commented 1 year ago

Great point. I think I can do a version check of pytorch before calling that function to avoid that issue.

kevinjohncutler commented 1 year ago

Ok, latest version (including pip versions of omnipose and cellpose-omni) have fixed this issue. I just use a try/except branch to catch it for any version lacking backends.