Training on 2 GPU's

Question

Training on 2 GPU's

israrbacha opened this issue 5 years ago · 3 comments

HI,
thanks for your amazing work, i want to train this model on two rtx 2080ti gpus, do you think that its easy with the existing code or i need to make changes to make it run on 2 gpus? thanks

Answer 1 · 2019-10-20T14:34:02.000Z

@israrbacha You need to make changes, but I am mot familiar with multiple GPU training of tensorflow 2.0. I will try when I am free.

Answer 2 · 2020-01-20T18:20:06.000Z

HI，
I want use your code to train the scanner image which is 16bit grayscale image, I will try to use Pydicom or Tensorflow IO module to read the dcm file. I want to know does you model can train the one channel 16bit Image without modification of the core module?
Thanks

Answer 3 · 2020-01-21T02:14:11.000Z

@qiuLimoges The easiest way I can think of is to convert the 16bit image to RGB image (three identical channels).