unet-lits-2d-pipeline
unet-lits-2d-pipeline copied to clipboard
GPU using
In the program, 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.