YogeLiu/TexTeller
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[deps] Change onnx-gpu to manually install

Browse Source
This commit is contained in:
OleehyO
2025-04-02 02:48:23 +00:00
parent 5673adecff
commit 509cb75dfa
2 changed files with 0 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
pyproject.toml
View File

@@ -18,7 +18,6 @@ dependencies = [
"evaluate>=0.4.3", "evaluate>=0.4.3",
"onnx>=1.17.0", "onnx>=1.17.0",
"onnxruntime>=1.20.1", "onnxruntime>=1.20.1",
"onnxruntime-gpu>=1.20.1 ; sys_platform == 'linux'",
"opencv-python>=4.11.0.86", "opencv-python>=4.11.0.86",
"optimum>=1.24.0", "optimum>=1.24.0",
"pyclipper>=1.3.0.post6", "pyclipper>=1.3.0.post6",

27
setup_platform.py
View File

@@ -1,27 +0,0 @@
#!/usr/bin/env python3
"""
Platform-specific setup script for TexTeller.
This script detects the platform and installs the appropriate version of onnxruntime.
"""
import platform
import subprocess
import sys
def install_platform_specific_dependencies():
"""Install the appropriate version of onnxruntime based on the platform."""
system = platform.system().lower()
if system == "linux":
print("Linux detected. Installing onnxruntime-gpu...")
subprocess.check_call([sys.executable, "-m", "pip", "install", "onnxruntime-gpu>=1.20.0"])
else:
print(f"{system.capitalize()} detected. Installing onnxruntime...")
subprocess.check_call([sys.executable, "-m", "pip", "install", "onnxruntime>=1.20.0"])
print("Platform-specific dependencies installed successfully.")
if __name__ == "__main__":
install_platform_specific_dependencies()