setup.py.in 887 字节
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',
11 12
          'paddle.v2.plot',
          'paddle.v2.master']
Z
zhangjinchao01 已提交
13

14 15 16 17 18 19 20 21 22
setup_requires=["requests",
                "numpy",
                "protobuf==3.1",
                "matplotlib",
                "rarfile"]

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

Z
zhangjinchao01 已提交
23 24 25
setup(name='paddle',
      version='${PADDLE_VERSION}',
      description='Parallel Distributed Deep Learning',
26
      install_requires=setup_requires,
L
Luo Tao 已提交
27
      packages=packages,
28
      package_data={'paddle.v2.master': ['libpaddle_master.so'], },
Z
zhangjinchao01 已提交
29 30
      package_dir={
          '': '${CMAKE_CURRENT_SOURCE_DIR}'
31
      },
Z
zhangjinchao01 已提交
32
)