search

ENHANCE-PET / MOOSE

MOOSE (Multi-organ objective segmentation) a data-centric AI solution that generates multilabel organ segmentations to facilitate systemic TB whole-person research.The pipeline is based on nn-UNet and has the capability to segment 120 unique tissue classes from a whole-body 18F-FDG PET/CT image.
https://enhance.pet
GNU General Public License v3.0
196 stars 28 forks source link

BUG: cuda accelerator is not recognized despite existing #77

Closed Keyn34 closed 1 year ago

Keyn34 commented 1 year ago

Description A potential error where moosez/torch cannot detect cuda. This happened twice, once on Windows and once on Ubuntu.

To Reproduce Steps to reproduce the behaviour:

  1. Create a virtual environment for moosez
  2. pip install the latest or specified moose version
  3. Prepare a dataset according to the moosez folder structure and run moosez on it
  4. If moosez/torch has trouble finding the correct cuda version, the error/warning will be displayed as below.

Expected behaviour moosez/torch recognizes the correct cuda version and determines the GPU as the prediction accelerator.

Screenshots This might be the error/warning. The prediction will run on the CPU afterwards.

image

System information:

Fix The fix for both was to reinstall torch with the corresponding command from https://pytorch.org/get-started/locally/

LalithShiyam commented 1 year ago

Hi @Keyn34, since the fix seems to solve the problem. I will leave it at that. Will add this to the FAQ - thanks!