Update launch_container.sh

launch_container.sh

+#/bin/bash
+
+# set -x
+# 脚本作用: 、
+# 1. bash launch.sh 直接进入容器
+# 2. bash launch.sh d / bash launch.sh destroy 删除容器
+# 3. bash launch.sh p / bash launch.sh print 打印容器名称
+
+
+
+docker_image=dcu_ai:v1.1
+export CONTAINER_NAME="yolov5_test_dtk24.04.1"
+
+command="docker run -id \
+    --name ${CONTAINER_NAME} \
+    --shm-size=32G \
+    --ipc=host \
+    --group-add video \
+    --cap-add=SYS_PTRACE \
+    --security-opt seccomp=unconfined \
+    --network=host \
+    --privileged \
+    -e https_proxy=http://10.17.26.123:8086 \
+    -e http_proxy=http://10.17.26.123:8086 \
+    --device=/dev/kfd \
+    --device=/dev/mkfd \
+    --device=/dev/dri \
+    -v /opt/hyhal:/opt/hyhal \
+    -v /home/wkx/yolov5/datasets:/datasets \
+    ${docker_image} \
+    /bin/bash"
+
+
+# 检查 CONTAINER_NAME 环境变量是否设置
+if [ -z "${CONTAINER_NAME}" ]; then
+    echo "环境变量 CONTAINER_NAME 未设置，请设置后再试。"
+    exit 1
+fi
+
+
+check_command_result() {
+    # 获取传递给函数的退出状态
+    local exit_status=$1
+    
+    # 根据退出状态采取行动
+    if [ "$exit_status" -eq 0 ]; then
+        echo "命令成功执行"
+    else
+        echo "命令执行失败，退出状态为: $exit_status"
+        exit $exit_status
+    fi
+}
+
+
+# 使用 docker ps -a | grep 来查找容器，并计算结果行数
+num=$(docker ps -a | grep "$CONTAINER_NAME" | wc -l)
+
+# 检查容器是否存在
+if [ "$num" -gt 0 ]; then
+    echo "Container:[${CONTAINER_NAME}] already exists"
+
+    #! 功能: 检测是否输入参数，而且参数是否符合类型，进行停止删除操作
+    # 检查是否有参数传入
+    if [ "$#" -gt 0 ]; then
+        # 获取第一个参数
+        param="$1"
+        if [ "$param" = "d" ] || [ "$param" = "destroy" ]; then
+            # echo "参数是 d 或者 destroy"
+            docker stop ${CONTAINER_NAME}
+            docker rm -f ${CONTAINER_NAME}
+            echo "container:[${CONTAINER_NAME}] finish destroy."
+        fi
+
+        if [ "$param" = "p" ] || [ "$param" = "print" ]; then
+            # echo "参数是 d 或者 destroy"
+            echo "container:[${CONTAINER_NAME}]."
+            exit
+        fi
+    else  #! 如果没参数传入, 则直接进入容器
+        docker start $CONTAINER_NAME        
+        docker exec -ti --env COLUMNS=`tput cols` --env LINES=`tput lines` $CONTAINER_NAME /bin/bash
+        exit
+    fi 
+    #!
+
+else
+    echo "Container:[${CONTAINER_NAME}] does not exist, will be created and go inside it."
+
+    eval ${command}
+
+    check_command_result $?
+    echo "Container:[${CONTAINER_NAME}] Created successfully!"
+
+    docker start $CONTAINER_NAME
+    exit_status=$?
+
+
+    echo "Container:[${CONTAINER_NAME}] Started successfully!"
+    docker exec -ti --env COLUMNS=`tput cols` --env LINES=`tput lines` $CONTAINER_NAME /bin/bash
+    exit
+fi