forked from moskomule/anatome
-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
40 lines (34 loc) · 1.21 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
"""
conda create -n uanatome python=3.9
conda activate uanatome
conda remove --all --name uanatome
rm -rf /Users/brando/anaconda3/envs/uanatome
"""
from setuptools import find_packages, setup
import pathlib
# The directory containing this file
# HERE = pathlib.Path(__file__).parent
# here = os.path.abspath(os.path.dirname(__file__))
HERE = pathlib.Path('~/ultimate-anatome/').expanduser()
# The text of the README file
README = (HERE / "README.md").read_text()
setup(
name='ultimate-anatome',
version='0.1.1',
description='Ἀνατομή (Anatome) is a PyTorch library to analyze representation of neural networks',
long_description=README,
long_description_content_type="text/markdown",
url='https://github.com/brando90/ultimate-anatome',
author='Brando Miranda',
author_email='brandojazz@gmail.com',
python_requires='>=3.9.0',
license='MIT',
packages=find_packages(),
# package_dir={'': 'anatome'},
# packages=find_packages('anatome'), # imports all modules/folders with __init__.py & python files
install_requires=['torch>=1.9.0',
'torchvision>=0.10.0',
'torchaudio>=0.9.1',
'tqdm'
]
)