Skip to content

Commit 7bfb390

Browse files
authored
Update runcellpose.py (#222)
1 parent 6a566d8 commit 7bfb390

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

active_plugins/runcellpose.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -673,6 +673,7 @@ def display(self, workspace, figure):
673673
def do_check_gpu(self):
674674
import importlib.util
675675
torch_installed = importlib.util.find_spec('torch') is not None
676+
self.cellpose_ver = importlib.metadata.version('cellpose')
676677
#if the old version of cellpose <2.0, then use istorch kwarg
677678
if float(self.cellpose_ver[0:3]) >= 0.7 and int(self.cellpose_ver[0])<2:
678679
GPU_works = core.use_gpu(istorch=torch_installed)

0 commit comments

Comments
 (0)