Feat: Allow user to select GPU to run moose

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

Feat: Allow user to select GPU to run moose #1

Closed LalithShiyam closed 1 year ago

LalithShiyam commented 2 years ago

Allow users to select the GPU to run moose. Makes a lot of sense if the user has a large server. Would be pivotal for multimoose.

@LalithShiyam to develop the prototype
@Keyn34 to test it

chris-clem commented 2 years ago

For now, you could let users know that they can do it with CUDA_VISIBLE_DEVICES=3 moose -f /home/user/Data/Patients_moose where CUDA_VISIBLE_DEVICES is the GPU number.

LalithShiyam commented 1 year ago

Cool! closing this as its clear now @chris-clem !