nnUNet
nnUNet copied to clipboard
About problems encountered when using multiple graphics cards
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
Hi 514tzy, Could you please run your code from a terminal and share the error message?