0xshipthecode/dl-docker

Docker image for deep learning with a few popular DL frameworks (TensorFlow, Chainer, etc.)

Modification of floydhub/dl-docker

Forked from floydhub and modified to support chainer. Ubuntu image uipgraded to 16.04 for CPU version.

See the original README.md.

Specs

This is what you get out of the box when you create a container with the provided image/Dockerfile:

• Ubuntu 16.04
• CUDA 8.0 (GPU version only)
• cuDNN v5 (GPU version only)
• Tensorflow
• Keras
• iPython/Jupyter Notebook (including iTorch kernel)
• Numpy, SciPy, Pandas, Scikit Learn, Matplotlib
• OpenCV
• Chainer
• A few common libraries used for deep learning