Skip to content

Commit

Permalink
fix python version
Browse files Browse the repository at this point in the history
  • Loading branch information
Hideousmon committed Oct 29, 2023
1 parent 57c29c3 commit 2fb171f
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 6 deletions.
2 changes: 1 addition & 1 deletion tools/pnnx/python/README.md
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
# pnnx
python wrapper of pnnx, only support python3.x now.
python wrapper of pnnx, only support python 3.7+ now.

Install from pip
==================
Expand Down
7 changes: 2 additions & 5 deletions tools/pnnx/setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -62,7 +62,6 @@ def build_extension(self, ext):
"-DCMAKE_RUNTIME_OUTPUT_DIRECTORY_RELEASE={}".format(extdir),
"-DPython3_EXECUTABLE={}".format(sys.executable),
"-DCMAKE_BUILD_TYPE={}".format(cfg), # not used on MSVC, but no harm
"-DPNNX_PYTHON=ON"
]

if TORCH_INSTALL_DIR != "":
Expand Down Expand Up @@ -111,7 +110,7 @@ def build_extension(self, ext):
# CMake 3.12+ only.
build_args += ["-j{}".format(self.parallel)]
else:
build_args += ["-j4"]
build_args += ["-j2"]

if not os.path.exists(self.build_temp):
os.makedirs(self.build_temp)
Expand Down Expand Up @@ -143,18 +142,16 @@ def build_extension(self, ext):
classifiers=[
"Programming Language :: C++",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.6",
"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",
"Programming Language :: Python :: 3.12",
"License :: OSI Approved :: BSD License",
"Operating System :: OS Independent",
],
license="BSD-3",
python_requires=">=3.6",
python_requires=">=3.7",
packages=find_packages("python"),
package_dir={"": "python"},
install_requires=requirements,
Expand Down

0 comments on commit 2fb171f

Please sign in to comment.