uoguelph-mlrg / Theano-MPI

MPI Parallel framework for training deep learning models built in Theano
Other
53 stars 21 forks source link

Does it work on one Cloud virtual machine with 8 gpus #28

Open wangluochao902 opened 5 years ago

wangluochao902 commented 5 years ago

I am new to MPI and Theano. I am trying to make it work but I haven't yes.

I set up a google cloud virtual machine with 8 GPUs. I am curious whether it is the right setting. It seems like we need 8 hosts and each host has one GPU. Does it mean I need to set up 8 virtual machines each with one GPU?

Thanks

wangluochao902 commented 5 years ago

I suspect it will work since I see the session_single_node.cfg file.