retry_env.sh 847 Bytes
Newer Older
yyy's avatar
yyy committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
#!/bin/bash

# 定义最大重试次数
max_retries=5
retry_count=0

while true; do
    # prepare env
    source activate MinerU
    pip install -r requirements-qa.txt
    pip install magic-pdf[full]==0.7.0b1 --extra-index-url https://wheels.myhloli.com -i https://pypi.tuna.tsinghua.edu.cn/simple
    pip install paddlepaddle-gpu==3.0.0b1 -i https://www.paddlepaddle.org.cn/packages/stable/cu118/
    exit_code=$?
    if [ $exit_code -eq 0 ]; then
        echo "test.sh 成功执行!"
        break
    else
        let retry_count+=1
        if [ $retry_count -ge $max_retries ]; then
            echo "达到最大重试次数 ($max_retries),放弃重试。"
            exit 1
        fi
        echo "test.sh 执行失败 (退出码: $exit_code)。尝试第 $retry_count 次重试..."
        sleep 5  # 等待 5 秒后重试
    fi
done