clks-wzz/FAS-SGTD

Encountered a problem while trying to train on multiple gpus.

Closed this issue · 2 comments

When I was trying to train the model on 1 or 2 gpus the code works well. But training on more than 2 gpus will cause error. Is that because of your design? Thank you so much.

Please show your settings of scripts and GPUs.

still a tf version issue. Should only use tensorflow 1.14 or tensorflow-gpu 1.14 if you want to train on GPU.