LeeDoYup/DeblurGAN-tf

GPU memory using

huyz1117 opened this issue · 1 comments

huyz1117 commented

Hi, thank you for sharing the code
When i run the code, it occupies all of the memory of 4 GPUs. But only one GPU is involved in the calculation. (my work computer has 4 1080Ti).
Could you reply the question?
Thank you

LeeDoYup commented

You can (manually) assign a GPU in tensorflow. If not, tensorflow occupy all memory.