I ran into trouble using the official Nvidia Digits docker build because it doesn't fully support the Pascal architecture used by the new Titan X.
So I combined information from the following sources to create this Dockerfile.
-
CUDA 8: https://github.com/NVIDIA/nvidia-docker/blob/master/ubuntu-14.04/cuda/8.0/runtime/Dockerfile
-
CAFFE: https://github.com/BVLC/caffe/blob/master/docker/standalone/gpu/Dockerfile
-
CAFFE & CUDA 8 ISSUES: NVIDIA/nvidia-docker#165
-
DIGITS 4: https://github.com/NVIDIA/nvidia-docker/blob/master/ubuntu-14.04/digits/4.0/Dockerfile