File size: 1,847 Bytes
15fe7bc a6bbecf 15fe7bc a6bbecf 15fe7bc a6bbecf 15fe7bc |
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 |
# 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.
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="[email protected]",
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",
"common/cudaraster/*.hpp",
"common/cudaraster/impl/*.cpp",
"common/cudaraster/impl/*.hpp",
"common/cudaraster/impl/*.inl",
"common/cudaraster/impl/*.cu",
"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",
)
|