You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
After clicking the "Test GPU" button in the RunCellpose module, this error shows up in my stdout:
Traceback (most recent call last):
File "/path/to/conda/env/lib/python3.8/site-packages/cellprofiler/gui/module_view/_module_view.py", line 1206, in callback
self.__on_do_something(event, setting)
File "/path/to/conda/env/lib/python3.8/site-packages/cellprofiler/gui/module_view/_module_view.py", line 2247, in __on_do_something
setting.on_event_fired()
File "/path/to/conda/env/lib/python3.8/site-packages/cellprofiler_core/setting/do_something/_do_something.py", line 27, in on_event_fired
self.__callback(*self.__args)
File "/path/to/CP.plugins/CellProfiler-plugins/active_plugins/runcellpose.py", line 724, in do_check_gpu
if self.save_probabilities.value:
NameError: name 'core' is not defined`
I can open a PR to patch this with the proposed change below:
Starting on current line 799 of runcellpose.py - add the import core statement
def do_check_gpu(self):
import importlib.util
from cellpose import core
torch_installed = importlib.util.find_spec("torch") is not None
self.cellpose_ver = importlib.metadata.version("cellpose")
# if the old version of cellpose <2.0, then use istorch kwarg
if float(self.cellpose_ver[0:3]) >= 0.7 and int(self.cellpose_ver[0]) < 2:
GPU_works = core.use_gpu(istorch=torch_installed)
else: # if new version of cellpose, use use_torch kwarg
GPU_works = core.use_gpu(use_torch=torch_installed)
if GPU_works:
message = "GPU appears to be working correctly!"
else:
message = (
"GPU test failed. There may be something wrong with your configuration."
)
import wx
wx.MessageBox(message, caption="GPU Test")
The text was updated successfully, but these errors were encountered:
After clicking the "Test GPU" button in the RunCellpose module, this error shows up in my stdout:
I can open a PR to patch this with the proposed change below:
Starting on current line 799 of
runcellpose.py
- add the import core statementThe text was updated successfully, but these errors were encountered: