unet-lits-2d-pipeline icon indicating copy to clipboard operation
unet-lits-2d-pipeline copied to clipboard

GPU use

Open lxg15066629402 opened this issue 5 years ago • 0 comments

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.

lxg15066629402 avatar Sep 25 '19 07:09 lxg15066629402