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

[eggs]
eggs +=
Julien Muchembled's avatar
Julien Muchembled committed
14 15
  ${tensorflow_tensorboard:egg}
  ${tensorflow:egg}
16 17 18
  ${protobuf-python:egg}
  ${h5py:egg}
  ${pillow-python:egg}
19 20 21
  ${python-PyYAML:egg}
  six
  keras
22 23 24
  filelock
  nose
  ${chainer:egg}
Julien Muchembled's avatar
Julien Muchembled committed
25 26
  ${torch:egg}
  ${pycocotools:egg}
27
  torchvision
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
35
tf_cudnn_version = 5.1.10
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