nnUNet icon indicating copy to clipboard operation
nnUNet copied to clipboard

About problems encountered when using multiple graphics cards

Open 514tzy opened this issue 9 months ago • 1 comments

I want to use 4 blocks 2080ti to train the brats2021 dataset, using the following two commands: export CUDA_VISIBLE_DEVICES=0,1,2,3 nnUNetv2_train 137 2d 0 -num_gpus 4 However, the project will exit pycahrm directly after running for a while. How can I solve this problem? thank you

514tzy avatar Apr 27 '24 07:04 514tzy

Hi 514tzy, Could you please run your code from a terminal and share the error message?

Kobalt93 avatar May 08 '24 13:05 Kobalt93