Skip to content

slapos
slapos
Commit 4ee4a6d8 authored by Yusei Tahara's avatar Yusei Tahara
Browse files
Options

wendelin: support keras 2.0.0 and tensorflow 1.0.1.

parent fddde5f6
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
software/wendelin/software-kerastensorflow.cfg
View file @ 4ee4a6d8
...@@ -11,7 +11,9 @@ eggs += ...@@ -11,7 +11,9 @@ eggs +=
${protobuf-python:egg} ${protobuf-python:egg}
${h5py:egg} ${h5py:egg}
${pillow-python:egg} ${pillow-python:egg}
Keras ${python-PyYAML:egg}
six
keras
[cuda] [cuda]
# If you use cuda, please adjust paramters for your environment # If you use cuda, please adjust paramters for your environment
...@@ -19,7 +21,7 @@ eggs += ...@@ -19,7 +21,7 @@ eggs +=
# https://developer.nvidia.com/cuda-gpus # https://developer.nvidia.com/cuda-gpus
tf_need_cuda = 1 tf_need_cuda = 1
tf_cuda_version = 8.0 tf_cuda_version = 8.0
tf_cudnn_version = 6.5 tf_cudnn_version = 5.1.10
tf_cuda_compute_capabilities = 5.2,6.1 tf_cuda_compute_capabilities = 5.2,6.1
cuda_toolkit_path = /usr/local/cuda cuda_toolkit_path = /usr/local/cuda
cudnn_install_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
GitLab Nexedi Edition | About GitLab | About Nexedi | 沪ICP备2021021310号-2 | 沪ICP备2021021310号-7