Init

Dockerfile

@@ -0,0 +1,17 @@
+FROM nvidia/cuda:8.0
+
+MAINTAINER Till von Ahnen "[email protected]"
+
+# prevent shell interactions
+ENV DEBIAN_FRONTEND noninteractive
+ENV DEBCONF_NONINTERACTIVE_SEEN true
+
+# prepare for pip3
+RUN apt-get update
+RUN apt-get install python3-pip python3 python3-dev -y
+
+# get the latest pip
+RUN pip3 install --upgrade pip
+
+# install everything needed
+RUN pip3 install tensorflow-gpu numpy scipy scikit-learn pillow h5py keras

README.md

@@ -1,2 +1,16 @@
 # keras-tf-gpu
 Docker image for keras using tensorflow-gpu
+
+It includes:
+
+* cuda 8.0
+* tensorflow 1.0
+* keras
+
+## Usage
+
+`nvidia-docker run --rm -ti xoryouyou/keras-tf-gpu bash`
+
+When using tensorboard just put it in the background for example:
+
+`tensorboard --log-dir=/tmp/logs &`