Commit 4ee4a6d8 authored by Yusei Tahara's avatar Yusei Tahara

wendelin: support keras 2.0.0 and tensorflow 1.0.1.

parent fddde5f6
......@@ -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
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment