Docker image based on NVIDIA/caffe, NVIDIA's fork of BVLC/caffe.
See the requirements for CUDA since the Caffe image is based on a cuDNN runtime image.
This image can be used as a base image for other images like [[DIGITS]], or it can be used directly to train networks:
# Run a Caffe training job
docker run --rm --runtime=nvidia nvidia/caffe caffe train --solver <args>Check the DockerHub