Example to use with MPI

NVIDIA / caffe

Caffe: a fast open framework for deep learning.

http://caffe.berkeleyvision.org/

Other

672 stars 263 forks source link

Example to use with MPI #505

Closed dashesy closed 6 years ago

dashesy commented 6 years ago

Is there any example to use NVIDIA/caffe with MPI? Using NCCL2, it should be possible I guess but it was not stated if MPI is supported. There is also https://github.com/Caffe-MPI/Caffe-MPI.github.io but I do not see why NVIDIA/caffe itself could not be used with MPI (and OpenIB)

Update: checking the code, MPI support is not implemented for NVIDIA/caffe, but comparign to Caffe-MPI, adding the support is not too difficult.

drnikolaev commented 6 years ago

@dashesy could you verify https://github.com/drnikolaev/caffe/tree/caffe-0.17 release candidate? It's different implementation though (closing your PR, thank you).

drnikolaev commented 6 years ago

@dashesy Please verify https://github.com/NVIDIA/caffe/tree/v0.17.1 release and reopen the issue if needed.