mirror of
https://github.com/yuruotong1/autoMate.git
synced 2025-12-25 21:06:47 +08:00
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
import subprocess
|
|
import sys
|
|
from util import download_weights
|
|
import urllib.request
|
|
import urllib.error
|
|
|
|
def install_requirements():
|
|
# Check if Google is accessible
|
|
try:
|
|
# Try to connect to Google with a timeout of 3 seconds
|
|
urllib.request.urlopen('https://www.google.com', timeout=3)
|
|
# If successful, install normally
|
|
subprocess.run([sys.executable, '-m', 'pip', 'install', '-r', 'requirements.txt'])
|
|
except (urllib.error.URLError, TimeoutError):
|
|
print("Using Tsinghua mirror")
|
|
subprocess.run([
|
|
sys.executable, '-m', 'pip', 'install',
|
|
'-r', 'requirements.txt',
|
|
'-i', 'https://pypi.tuna.tsinghua.edu.cn/simple'
|
|
])
|
|
|
|
|
|
def adjust_python_env():
|
|
# check if python is 3.12
|
|
if sys.version_info.major != 3 or sys.version_info.minor != 12:
|
|
print("Python version is not 3.12, please install python 3.12")
|
|
exit(1)
|
|
|
|
def install():
|
|
adjust_python_env()
|
|
install_requirements()
|
|
# download the weight files
|
|
download_weights.download()
|
|
print("Installation complete!")
|
|
|
|
if __name__ == "__main__":
|
|
install()
|
|
print("Installation complete!") |