[chore] Setup project infrastructure

This commit is contained in:
三洋三洋
2025-02-28 19:35:59 +08:00
parent 52fce4d39d
commit a8a005ae10
5 changed files with 268 additions and 25 deletions

27
setup_platform.py Normal file
View File

@@ -0,0 +1,27 @@
#!/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()