yiminglin/dl-image repository overview

⁠Supported software:

1. Python 2.7.12
2. CUDA 8.0 + CUDNN 5.1
3. OpenCV 3.1 (built from source with GPU and ffmpeg supports)
4. Caffe 1.0
5. Tensorflow 1.1 (Keras included)
6. Caffe2
7. TensorLayer⁠
8. PyTorch
9. Torch
10. MXNET
11. Scikit learn
12. Dlib 19.4

⁠To run this image with GPU support

on your machine:

• Install CUDA 8.0 and CUDNN 5.1+
• Install Docker⁠
• Install nvidia-docker plugin⁠

⁠Usage examples

• run the image:
  nvidia-docker run -it yiminglin/dl-image:latest

• mount the home directory of the host machine:
  nvidia-docker run -v /home/yiming/:$(pwd) -it yiminglin/dl-image:latest

• work with Jupyter notebook: nvidia-docker run -v /home/yiming/:$(pwd) -it -p 8888:8888 yiminglin/dl-image:latest sh -c "jupyter notebook --allow-root --no-browser --ip 0.0.0.0 /home/yiming"

⁠Tutorials in Chinese

• Docker--深度学习环境配置一站式解决方案 - 知乎专栏⁠
• Docker深度学习环境篇第二弹：PyCharm + Docker + GPUs = Everything you need for DL development⁠