forked from chainer/chainer
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
100 lines (91 loc) · 2.85 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
#!/usr/bin/env python
import sys
from setuptools import setup
import chainer_setup_build
setup_requires = []
install_requires = [
'filelock',
'nose',
'numpy>=1.9.0',
'protobuf',
'six>=1.9.0',
]
# Hack for Read the Docs
on_rtd = chainer_setup_build.check_readthedocs_environment()
if on_rtd:
print('Add develop command for Read the Docs')
sys.argv.insert(1, 'develop')
setup_requires = ['Cython>=0.23'] + setup_requires
chainer_setup_build.parse_args()
setup(
name='chainer',
version='1.16.0',
description='A flexible framework of neural networks',
author='Seiya Tokui',
author_email='[email protected]',
url='http://chainer.org/',
license='MIT License',
packages=['chainer',
'chainer.dataset',
'chainer.datasets',
'chainer.functions',
'chainer.functions.activation',
'chainer.functions.array',
'chainer.functions.caffe',
'chainer.functions.connection',
'chainer.functions.evaluation',
'chainer.functions.loss',
'chainer.functions.math',
'chainer.functions.noise',
'chainer.functions.normalization',
'chainer.functions.pooling',
'chainer.functions.util',
'chainer.function_hooks',
'chainer.iterators',
'chainer.initializers',
'chainer.links',
'chainer.links.activation',
'chainer.links.caffe',
'chainer.links.caffe.protobuf2',
'chainer.links.caffe.protobuf3',
'chainer.links.connection',
'chainer.links.loss',
'chainer.links.model',
'chainer.links.normalization',
'chainer.optimizers',
'chainer.serializers',
'chainer.testing',
'chainer.training',
'chainer.training.extensions',
'chainer.training.triggers',
'chainer.utils',
'cupy',
'cupy.binary',
'cupy.core',
'cupy.creation',
'cupy.cuda',
'cupy.indexing',
'cupy.io',
'cupy.linalg',
'cupy.logic',
'cupy.manipulation',
'cupy.math',
'cupy.padding',
'cupy.random',
'cupy.sorting',
'cupy.statistics',
'cupy.testing'],
package_data={
'cupy': ['core/carray.cuh'],
},
zip_safe=False,
setup_requires=setup_requires,
install_requires=install_requires,
tests_require=['mock',
'nose'],
# To trick build into running build_ext
ext_modules=[chainer_setup_build.dummy_extension],
cmdclass={
'build_ext': chainer_setup_build.chainer_build_ext,
},
)