Updated readme to keras 2.0; No Dockerfile changes needed.

xoryouyou · Mar 15, 2017 · 81a3a84 · 81a3a84
1 parent 6299d0b
commit 81a3a84
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -1,16 +1,18 @@
 # keras-tf-gpu
-Docker image for keras using tensorflow-gpu
+Docker image for keras using tensorflow-gpu based on ubuntu 16.04
 
 It includes:
 
 * cuda 8.0
+* cudnn 5.0
 * tensorflow 1.0
-* keras
+* keras 2.0
+
 
 ## 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 &`
+`tensorboard --logdir=/tmp/logs &`