How to use multiple GPU device?

[h2oai]

How to use multiple GPU device?

[mdymczyk]

@aspigex which network are you building? If you're not specifying the network type explicitly then the type will be inferred automatically (based on the input data format). You should be able to find it in the logs (lenet, alexnet, MLP, resnet, inception, vgg).

Currently only lenet is implemented in a multi-GPU fashion. For other networks you'd have to either contribute to deepwater by rewriting the network definition files (found here https://github.com/h2oai/deepwater/tree/master/tensorflow/src/main/resources/deepwater/models) to use multiple GPUs (like this one https://github.com/h2oai/deepwater/blob/master/tensorflow/src/main/resources/deepwater/models/lenet.py).

Or you can write your own multi-GPU TF network in Python, save the model meta file and use it in DeepWater. Here's an example how to do it https://github.com/h2oai/h2o-3/blob/master/examples/deeplearning/notebooks/deeplearning_tensorflow_cat_dog_mouse_lenet.ipynb (Custom models section).