deltaMic/setup.py at main · VirtualEmbryo/deltaMic · GitHub

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
from torch.utils.cpp_extension import BuildExtension, CppExtension, CUDAExtension
from setuptools import setup, find_packages
from pathlib import Path

this_directory = Path(__file__).parent
readme = (this_directory / "README.md").read_text()

common_kwargs = {
    'name': 'deltamic',
    'version': '0.1.0',
    'description': 'Differentiable rendering through a 3d fluorescence microscope',
    'url': 'https://github.com/sacha-ichbiah/deltamic',
    'author': 'Sacha Ichbiah',
    'author_email': 'sacha.ichbiah@polytechnique.org',
    'license': 'BSD',
    'include_package_data': True,
    'packages': find_packages(),
    'install_requires': ['numpy>=1.21.6',
                         'torch>=1.13.1',
                         'largesteps>=0.2.1',
                         'robust_laplacian>=0.2.2',
                         'scipy>=1.5.4',
                         'trimesh>=3.10.2'],
    'long_description': readme,
    'long_description_content_type': "text/markdown",
    'classifiers': [
        "Development Status :: 3 - Alpha",
        "Intended Audience :: Developers",
        "License :: OSI Approved :: BSD License",
        "Operating System :: OS Independent",
        "Programming Language :: Python",
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.8",
        "Programming Language :: Python :: 3.9",
        "Programming Language :: Python :: 3.10",
    ],
}

if not torch.cuda.is_available():
    setup(**common_kwargs)
else:
    setup(
        ext_modules=[
            CUDAExtension(
                'deltamic.fourier3d_cpp',
                sources=['deltamic/cpp/fourier3d_cuda.cpp', 'deltamic/cpp/fourier3d_cuda_kernel.cu'],
                extra_compile_args={'cxx': ['-g'], 'nvcc': ['-O2']}),
        ],
        cmdclass={'build_ext': BuildExtension},
        **common_kwargs
    )