search

kevinjohncutler / omnipose

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

Problem using GPU #38

Open koopa31 opened 1 year ago

koopa31 commented 1 year ago

Hi, I've been trying to use Omnipose with GPU for two days. Actually, Cellpose works like charm, but when I try to use Omnipose I always get this error: RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED. When I do torch.cuda.isavailable() I get "True", so I don't understand. I tried to remove and re install torch with cuda, several times with different versions, but I always get the same error. Could you help me please? Thank you.

koopa31 commented 1 year ago

Actually, I succeeded in making it work on Windows, but I keep getting the same problem on Ubuntu 22.04. Could that be related to my Linux version?

kevinjohncutler commented 1 year ago

@koopa31 Yes, it must be something to do with your system, drivers, and torch version. What is your GPU and cuda/driver info? The only thing I can think of that might allow Cellpose to work but not Omnipose is that I could have made the install dependencies for Omnipose too new to work on your system. Some older graphics cards will not work with newer versions of pytorch (torch needs more recent drivers). A quick check is to compare the conda list output in your cellpose and omnipose environments.