You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

setup.py.i 1.5 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. import sys
  2. from setuptools import setup, find_packages
  3. try:
  4. from wheel.bdist_wheel import bdist_wheel as _bdist_wheel
  5. class bdist_wheel(_bdist_wheel):
  6. def finalize_options(self):
  7. _bdist_wheel.finalize_options(self)
  8. self.root_is_pure = False
  9. except ImportError:
  10. bdist_wheel = None
  11. if sys.version_info < (3, 0):
  12. sys.exit("Sorry, Python < 3.0 is not supported")
  13. requirements = ["torch"]
  14. setup(
  15. name="pnnx",
  16. version="${PACKAGE_VERSION}",
  17. author="nihui",
  18. author_email="nihuini@tencent.com",
  19. description="pnnx is an open standard for PyTorch model interoperability.",
  20. url="https://github.com/Tencent/ncnn/tree/master/tools/pnnx",
  21. classifiers=[
  22. "Programming Language :: C++",
  23. "Programming Language :: Python :: 3",
  24. "Programming Language :: Python :: 3.7",
  25. "Programming Language :: Python :: 3.8",
  26. "Programming Language :: Python :: 3.9",
  27. "Programming Language :: Python :: 3.10",
  28. "Programming Language :: Python :: 3.11",
  29. "License :: OSI Approved :: BSD License",
  30. "Operating System :: OS Independent",
  31. ],
  32. license="BSD-3",
  33. python_requires=">=3.6",
  34. packages=find_packages(),
  35. package_dir={"": "."},
  36. package_data={"pnnx": ["pnnx${PYTHON_MODULE_PREFIX}${PYTHON_MODULE_EXTENSION}"]},
  37. install_requires=requirements,
  38. cmdclass={"bdist_wheel": bdist_wheel},
  39. )