Multi GPU for Caffe - Githubissues

amplab / SparkNet

Distributed Neural Networks for Spark

MIT License

603 stars 172 forks source link

Multi GPU for Caffe #127

Open pcmoritz opened 8 years ago

pcmoritz commented 8 years ago

This is an issue for tracking the status of implementing https://github.com/amplab/SparkNet/commit/b73ed9780a1a21c5a1f2068bc6cb017857ee6f49 on top of the current JavaCPP architecture.

TODOs for completing this:

[x] Integrate the patch for caffe/include/caffe/parallel.hpp and caffe/src/caffe/parallel.cpp into the repo https://github.com/amplab/caffe
[ ] Build up to date JARs for this version of caffe
[ ] Create a MultiGPU example