From 81a3a84fe877b6c7915051ca9f5e3d868eb0f14f Mon Sep 17 00:00:00 2001 From: xoryouyou Date: Wed, 15 Mar 2017 10:15:05 +0100 Subject: [PATCH] Updated readme to keras 2.0; No Dockerfile changes needed. --- README.md | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 7544dcb..85f9422 100644 --- a/README.md +++ b/README.md @@ -1,11 +1,13 @@ # 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 @@ -13,4 +15,4 @@ It includes: When using tensorboard just put it in the background for example: -`tensorboard --log-dir=/tmp/logs &` +`tensorboard --logdir=/tmp/logs &`