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