Files
ncnn/tools/pnnx/python/setup.py.i
T

45 lines
1.5 KiB
OpenEdge ABL

import sys
from setuptools import setup, find_packages
try:
from wheel.bdist_wheel import bdist_wheel as _bdist_wheel
class bdist_wheel(_bdist_wheel):
def finalize_options(self):
_bdist_wheel.finalize_options(self)
self.root_is_pure = False
except ImportError:
bdist_wheel = None
if sys.version_info < (3, 0):
sys.exit("Sorry, Python < 3.0 is not supported")
requirements = ["torch"]
setup(
name="pnnx",
version="${PACKAGE_VERSION}",
author="nihui",
author_email="nihuini@tencent.com",
description="pnnx is an open standard for PyTorch model interoperability.",
url="https://github.com/Tencent/ncnn/tree/master/tools/pnnx",
classifiers=[
"Programming Language :: C++",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.7",
"Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"License :: OSI Approved :: BSD License",
"Operating System :: OS Independent",
],
license="BSD-3",
python_requires=">=3.6",
packages=find_packages(),
package_dir={"": "."},
package_data={"pnnx": ["pnnx${PYTHON_MODULE_PREFIX}${PYTHON_MODULE_EXTENSION}"]},
install_requires=requirements,
cmdclass={"bdist_wheel": bdist_wheel},
)