Public | Automated Build

Last pushed: a year ago
Short Description
Nvidia cuda + cudnn + spark
Full Description

CUDNN-SPARK

This dockerfile is destined for research and development of machine learning technologies especially deep learning. It can be use with mesos cluster.
cudnn-spark is based on nvidia/cuda:cudnn and consists of spark

Requirements

Usage

Local - spark-submit

nvidia-docker run --net host kbokh/cudnn-spark spark-submit \
       --master mesos://<mesos-master-host>:<port> \
       --deploy-mode client \
       --conf spark.mesos.executor.docker.image=kbokh/cudnn-spark
       --conf spark.mesos.mesosExecutor.cores=<num of cores> \
       --conf spark.mesos.coarse=false \
       --conf spark.mesos.executor.docker.volumes=/var/lib/mesos:/var/lib/mesos:rw \
       --class <main-class> /<app-jar>

where <main-class> /<app-jar> is a spark application

Local - spark-shell

nvidia-docker run --net host kbokh/cudnn-spark spark-shell

Remote - marathon

TODO - Add marathon json
Docker Pull Command
Owner
krot
Source Repository