An API for hosted deep learning models

Re:An API for hosted deep learning models by dharma1 on Saturday July 16, 2016 @02:23AM (#12105459)

Thanks for your superb work. Is it possible to use nvidia-docker on several AWS instances, to use multiple GPUs? (To spread training on multiple GPUs for more speed and ram. Tensorflow and Caffe support distributed training but not sure if it's viable on dockerized envs on AWS?)

[ Parent | on HN ]

Re:An API for hosted deep learning models by flx42_ on Saturday July 16, 2016 @02:30AM (#12105476)

One container can use multiple GPUs on the same machine without problems.

For distributed training (which Caffe doesn't actually support, not the official version), you would have to run one container per instance, but this is more a configuration problem at the framework level, than a Docker or nvidia-docker problem.

[ Parent | on HN ]