setup.py.in 1.3 KB
Newer Older
Z
zhangjinchao01 已提交
1 2 3 4 5 6
from setuptools import setup

packages=['paddle',
          'paddle.proto',
          'paddle.trainer',
          'paddle.trainer_config_helpers',
Q
qiaolongfei 已提交
7
          'paddle.utils',
Y
Yu Yang 已提交
8 9
          'paddle.v2',
          'paddle.v2.dataset',
Y
Yancey1989 已提交
10
          'paddle.v2.reader',
H
Helin Wang 已提交
11
          'paddle.v2.master',
12 13 14
          'paddle.v2.plot',
          'paddle.v2.framework',
          'paddle.v2.framework.proto']
Z
zhangjinchao01 已提交
15

16
setup_requires=["requests",
Y
Yu Yang 已提交
17
                "numpy>=1.12",
18
                "protobuf==3.1",
G
gongweibao 已提交
19
                "recordio",
20
                "matplotlib",
21
                "rarfile",
22
                "scipy>=0.19.0",
23
                "Pillow",
Y
Yu Yang 已提交
24
                "nltk"]
25 26 27 28

if '${CMAKE_SYSTEM_PROCESSOR}' not in ['arm', 'armv7-a', 'aarch64']:
    setup_requires+=["opencv-python"]

Z
zhangjinchao01 已提交
29 30 31
setup(name='paddle',
      version='${PADDLE_VERSION}',
      description='Parallel Distributed Deep Learning',
32
      install_requires=setup_requires,
L
Luo Tao 已提交
33
      packages=packages,
34 35 36
      package_data={'paddle.v2.master': ['libpaddle_master.so'],
            'paddle.v2.framework': ['core.so']
      },
Z
zhangjinchao01 已提交
37
      package_dir={
38 39 40
          '': '${CMAKE_CURRENT_SOURCE_DIR}',
          # The paddle.v2.framework.proto will be generated while compiling.
          # So that package points to other directory.
41
          'paddle.v2.framework.proto': '${PROJ_BINARY_ROOT}/paddle/framework'
42
      },
Z
zhangjinchao01 已提交
43
)