[deps] Change onnx-gpu to manually install
This commit is contained in:
@@ -18,7 +18,6 @@ dependencies = [
|
||||
"evaluate>=0.4.3",
|
||||
"onnx>=1.17.0",
|
||||
"onnxruntime>=1.20.1",
|
||||
"onnxruntime-gpu>=1.20.1 ; sys_platform == 'linux'",
|
||||
"opencv-python>=4.11.0.86",
|
||||
"optimum>=1.24.0",
|
||||
"pyclipper>=1.3.0.post6",
|
||||
|
||||
@@ -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()
|
||||
Reference in New Issue
Block a user