-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
59 lines (55 loc) · 1.81 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
import os
from setuptools import setup, find_packages
# Get CUDA version from the env variable, defaulting to '11.6' if not set
cuda_version = os.getenv('CUDA_VERSION', '11.6')
# Strip the period from the version string (e.g., '11.6' -> '116')
cuda_version = cuda_version.replace('.', '')
# Format version to get prebuilt wheel
cupy_package = f'cupy-cuda{cuda_version}'
setup(
name='oct_vesselseg',
version='0.0.7',
description='A Label-Free and Data-Free Synthesis Engine and Training Framework for Vascular Segmentation of sOCT Data with PyTorch.',
long_description=open('README.md').read(),
long_description_content_type="text/markdown",
author='Etienne Chollet & Yaël Balbastre',
author_email='etiennepchollet@gmail.com',
project_urls={
"Conference Preprint": "https://arxiv.org/abs/2405.13757v1",
"Paper Preprint": "https://arxiv.org/abs/2407.01419v1",
"Source": "https://github.com/EtienneChollet/oct_vesselseg"
},
packages=find_packages(),
entry_points={
'console_scripts': [
'oct_vesselseg=oct_vesselseg:main.app'
]
},
install_requires=[
cupy_package,
'cppyy~=2.3',
'torch',
'torchvision',
'torchaudio',
'torchmetrics',
'jitfields',
'torch-interpol',
'torch-distmap',
'nibabel',
'pytorch-lightning',
'scikit-learn',
'matplotlib',
'tensorboard',
'pandas',
'cyclopts'
'cornucopia',
],
classifiers=[
'Development Status :: 4 - Beta',
'Intended Audience :: Science/Research',
'Topic :: Software Development :: Build Tools',
'License :: OSI Approved :: MIT License',
'Programming Language :: Python :: 3.9',
],
python_requires='~=3.9',
)