diff --git a/software/wendelin/software-kerastensorflow.cfg b/software/wendelin/software-kerastensorflow.cfg
index 4f6a0c4e7ff89db92bfaa0cf8b8acaa4bef7561f..206f0b01614f949e33e327679fe4736ba6d84f3a 100644
--- a/software/wendelin/software-kerastensorflow.cfg
+++ b/software/wendelin/software-kerastensorflow.cfg
@@ -11,7 +11,9 @@ eggs +=
   ${protobuf-python:egg}
   ${h5py:egg}
   ${pillow-python:egg}
-  Keras
+  ${python-PyYAML:egg}
+  six
+  keras
 
 [cuda]
 # If you use cuda, please adjust paramters for your environment
@@ -19,7 +21,7 @@ eggs +=
 # https://developer.nvidia.com/cuda-gpus
 tf_need_cuda = 1
 tf_cuda_version = 8.0
-tf_cudnn_version = 6.5
+tf_cudnn_version = 5.1.10
 tf_cuda_compute_capabilities = 5.2,6.1
 cuda_toolkit_path = /usr/local/cuda
 cudnn_install_path = /usr/local/cuda