How to support multi GPU trainning ?

[CarlHuangNuc]

Dear Author,

        Whether this base code can support Multi GPU training ??

I try to change this paramters ( parser.add_argument('--GPU', required=False, type=int, default=-1, help='ID of the GPU to use' )),, but only can using 1 GPU.

[SilvioGiancola]

I don't think multi-GPU was implemented, you can easily train with a single commodity GPU. --GPU is meant for the ID of the GPU to train on, in case you have more than one. If you have suggestions for multi-GPU training, feel free to make changes and do a pull request.

[CarlHuangNuc]

Thanks for your quick response. whether one Nvdia V100 GPU can run trainning process ?

[SilvioGiancola]

I think a v100 will be enough, yes. @heldJan can confirm :)

[heldJan]

Yes, a v100 is enough :)