unet-lits-2d-pipeline
unet-lits-2d-pipeline copied to clipboard
GPU use
I want to run the program with two GPUs. I fixed os.environ['CUDA_LAUNCH_BLOCKING'] = "0, 1" in Train_Networks but it is still unsuccessful. I would like to ask how to do this.