wangruowen/tensorflow repository overview

⁠TensorFlow + Keras Docker with GPU support

Based on tensorflow/tensorflow:1.13.1-gpu-py3-jupyter⁠. Installed the following packages:

keras==2.2.4

pandas==0.24.1

⁠How to run

1. Need to install Nvidia GPU driver: Only GPU driver is needed. No need to install CUDA⁠
2. Install nvidia-docker2: https://github.com/NVIDIA/nvidia-docker⁠
3. Run docker:

docker run --runtime=nvidia -it -p 8888:8888 --rm -u $(id -u):$(id -g) -v $PWD:/tf wangruowen/tensorflow:1.13.1-gpu-py3-jupyter bash

1. Run Jupyter Notebook (skip token/password for easy access):

$ # Inside docker container
$ jnote