software-kerastensorflow.cfg 692 Bytes
Newer Older
1 2 3
[buildout]
extends =
  ../../component/keras/buildout.cfg
4
  ../../component/chainer/buildout.cfg
5 6 7
  software.cfg
parts +=
  keras-egg
8
  chainer-egg
9 10 11 12 13 14 15

[eggs]
eggs +=
  ${tensorflow-build-install-egg:egg}
  ${protobuf-python:egg}
  ${h5py:egg}
  ${pillow-python:egg}
16 17 18
  ${python-PyYAML:egg}
  six
  keras
19 20 21
  filelock
  nose
  ${chainer:egg}
22 23 24 25 26 27 28

[cuda]
# If you use cuda, please adjust paramters for your environment
# else ignore this part.
# https://developer.nvidia.com/cuda-gpus
tf_need_cuda = 1
tf_cuda_version = 8.0
29
tf_cudnn_version = 5.1.10
30 31 32 33 34 35 36
tf_cuda_compute_capabilities = 5.2,6.1
cuda_toolkit_path = /usr/local/cuda
cudnn_install_path = /usr/local/cuda

[versions]
mock = 2.0.0
funcsigs = 1.0.2