Blame view

software/wendelin/software-kerastensorflow.cfg 875 Bytes
Yusei Tahara committed
1 2 3
[buildout]
extends =
  ../../component/keras/buildout.cfg
Yusei Tahara committed
4
  ../../component/chainer/buildout.cfg
Yusei Tahara committed
5
  ../../component/pytorch/buildout.cfg
Yusei Tahara committed
6 7 8
  software.cfg
parts +=
  keras-egg
Yusei Tahara committed
9
  chainer-egg
Yusei Tahara committed
10
  pytorch-egg
Yusei Tahara committed
11 12 13

[eggs]
eggs +=
Yusei Tahara committed
14
  ${tensorboard-build-install-egg:egg}
Yusei Tahara committed
15 16 17 18
  ${tensorflow-build-install-egg:egg}
  ${protobuf-python:egg}
  ${h5py:egg}
  ${pillow-python:egg}
Yusei Tahara committed
19 20 21
  ${python-PyYAML:egg}
  six
  keras
Yusei Tahara committed
22 23 24
  filelock
  nose
  ${chainer:egg}
Yusei Tahara committed
25 26 27
  ${pytorch-build-install-egg:egg}
  ${python-cocoapi-build-install-egg:egg}
  torchvision
Yusei Tahara committed
28 29 30 31 32 33 34

[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
Yusei Tahara committed
35
tf_cudnn_version = 5.1.10
Yusei Tahara committed
36 37 38 39 40 41 42
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