gwang-kim's picture
u
f12ab4c
# Copyright (c) 2020, NVIDIA CORPORATION. All rights reserved.
#
# NVIDIA CORPORATION and its licensors retain all intellectual property
# and proprietary rights in and to this software, related documentation
# and any modifications thereto. Any use, reproduction, disclosure or
# distribution of this software and related documentation without an express
# license agreement from NVIDIA CORPORATION is strictly prohibited.
from pose_estimation.nvdiffrast import nvdiffrast
import setuptools
import os
with open("README.md", "r") as fh:
long_description = fh.read()
setuptools.setup(
name="nvdiffrast",
version=nvdiffrast.__version__,
author="Samuli Laine",
author_email="slaine@nvidia.com",
description="nvdiffrast - modular primitives for high-performance differentiable rendering",
long_description=long_description,
long_description_content_type="text/markdown",
url="https://github.com/NVlabs/nvdiffrast",
packages=setuptools.find_packages(),
package_data={
'nvdiffrast': [
'common/*.h',
'common/*.inl',
'common/*.cu',
'common/*.cpp',
'lib/*.h',
'torch/*.h',
'torch/*.inl',
'torch/*.cpp',
'tensorflow/*.cu',
] + (['lib/*.lib'] if os.name == 'nt' else [])
},
include_package_data=True,
install_requires=['numpy'], # note: can't require torch here as it will install torch even for a TensorFlow container
classifiers=[
"Programming Language :: Python :: 3",
"Operating System :: OS Independent",
],
python_requires='>=3.6',
)