DeepSpark

Deeplearning framework running on Spark

Requires

• Apache Hadoop YARN and Spark
• Caffe prerequisites for Worker nodes (http://caffe.berkeleyvision.org/installation.html)
• CUDA
• cuDNN
• protobuf, glog, gflags

Build

$ mvn package

running demos

Sample running scripts are on scripts/

• runCIFAR.sh: CIFAR10 demo
• runImageNet.sh: imagenet demo
• modelConvert.sh: convert DeepSpark weight to caffemodel