forked from google/svcca
-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
34 lines (28 loc) · 1.14 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
#!/usr/bin/env python
from distutils.core import setup
import setuptools
import os
root_dir = os.path.abspath(os.path.dirname(__file__))
with open(f'{root_dir}/README.md') as f:
readme = f.read()
with open(f'{root_dir}/requirements.txt') as f:
requirements = f.read().split()
packages = setuptools.find_packages('.', include='svcca.*')
setup(name='svcca',
version='0.0.1',
description='SVCCA on Numpy, Cupy, and PyTorch',
long_description=readme,
author='Rasmus Diederichsen',
author_email='rasmus@peltarion.com',
url='https://github.com/themightyoarfish/svcca-gpu',
classifiers=['Intended Audience :: Science/Research',
'Topic :: Scientific/Engineering :: Visualization',
'Programming Language :: Python :: 3.6',
'License :: OSI Approved :: Apache License',
'Intended Audience :: Developers',
],
keywords='deep-learning pytorch cupy numpy svcca neural-networks machine-learning'.split(),
install_requires=requirements,
packages=packages,
zip_safe=False, # don't install egg, but source
)