setup.py

import os
from setuptools import find_packages

try:
    from skbuild import setup
except ImportError:
    raise ImportError('scikit-build is required for installing')

try:
    import torch
    torch_root = os.path.dirname(torch.__file__)
    torch_ver = torch.__version__.replace('.', '')

    if "USE_CUDA" in os.environ:
        try:
            use_cuda = int(os.environ["USE_CUDA"])
        except ValueError:
            use_cuda = os.environ["USE_CUDA"].lower() == "true"
    else:
        use_cuda = torch.cuda.is_available()
except ImportError:
    torch_root = torch_ver = ''
    use_cuda = False
    print("Pytorch not found, pytorch related operators will not be built.")

def full_version(): # get 
    from subprocess import check_output
    from setuptools_scm.version import get_local_node_and_date

    if use_cuda:
        try:
            cuda_ver = check_output(["nvcc", "-V"]).decode()
            ver_aidx = cuda_ver.find("release")
            ver_bidx = cuda_ver.find(',', ver_aidx)
            cuda_ver = cuda_ver[ver_aidx+8:ver_bidx].replace('.', '')
        except FileNotFoundError:
            cuda_ver = ''
    else:
        cuda_ver = ''

    full_ver = 'th' + torch_ver if torch_ver else ''
    full_ver += '.cu' + cuda_ver if cuda_ver else ''

    def full_scheme(version):
        ver_str = get_local_node_and_date(version)
        if full_ver:
            if ver_str.find("+") < 0:
                ver_str += "+" + full_ver
            else:
                ver_str += "." + full_ver
        return ver_str

    return {'local_scheme': full_scheme}

extras = {
    'doc': ['sphinx', 'recommonmark', 'sphinx_rtd_theme'],
    'test': ['pytest']
}

setup(
    name="d3d",
    use_scm_version=full_version,
    description="Customized tools for 3D object detection",
    long_description='(see project homepage)',
    author='Jacob Zhong',
    author_email='cmpute@gmail.com',
    url='https://github.com/cmpute/d3d',
    download_url='https://github.com/cmpute/d3d/archive/master.zip',
    license='BSD-3-Clause',
    packages=find_packages(),
    install_requires=['numpy>=1.17.0', 'scipy>=1.4', 'addict', 'pillow', 'tqdm', 'msgpack', 'filterpy', 'sortedcontainers'],
    setup_requires=['scikit-build', 'setuptools_scm', 'cython>=0.29.16'],
    extras_require=extras,
    classifiers=[
        'Programming Language :: C++',
        'Programming Language :: Cython',
        'Programming Language :: Python :: 3',
        'License :: OSI Approved :: BSD License',
        'Operating System :: OS Independent',
        'Development Status :: 2 - Pre-Alpha',
        'Topic :: Scientific/Engineering'
    ],
    keywords=['detection', '3d'],
    entry_points={
        'console_scripts': [
            'd3d_waymo_convert = d3d.dataset.waymo.converter:main',
            'd3d_nuscenes_convert = d3d.dataset.nuscenes.converter:main',
            'd3d_kitti_parse_result = d3d.dataset.kitti.object:parse_detection_output',
        ],
    },
    cmake_args=[
        '-DBUILD_WITH_PYTORCH=ON',
        '-DCMAKE_PREFIX_PATH=%s' % torch_root,
        '-DBUILD_WITH_CUDA=%s' % ("ON" if use_cuda else "OFF"),
        '-DCYTHON_WITH_NO_DOCSTRINGS_ARG=FALSE',
    ] if torch_root else [
        '-DBUILD_WITH_PYTORCH=OFF',
        '-DBUILD_WITH_CUDA=OFF'
    ]
)