Commit 4da20266 authored by wangkx1's avatar wangkx1
Browse files

Update launch_container.sh

parent aa8955ca
#!/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 打印容器名称
......@@ -9,20 +9,18 @@
# ==================== 配置区域 ====================
# 镜像配置
docker_image="dcu_ai:v1.1"
container_name="yolov5_test_dtk24.04.1"
docker_image="harbor.sourcefind.cn:5443/dcu/admin/base/vllm:0.11.0-ubuntu22.04-dtk25.04.2-1226-das1.7-py3.10-20251226"
container_name="wkx-dtk25042-vllm-0.11.0-1226"
# 挂载卷配置 (格式: "宿主机路径:容器路径")
# 可以添加多个挂载,用空格分隔
mounts=(
"/opt/hyhal:/opt/hyhal"
"/home/wkx/yolov5/datasets:/datasets"
"/datav:/datav"
)
# 环境变量配置
environment_vars=(
"https_proxy=http://10.17.26.123:8086"
"http_proxy=http://10.17.26.123:8086"
)
# Docker 设备配置
......@@ -50,27 +48,27 @@ export CONTAINER_NAME="${container_name}"
# 构建 docker run 命令
build_docker_command() {
local cmd="docker run -id --name ${CONTAINER_NAME} --shm-size=${shm_size}"
# 添加额外参数
for arg in "${extra_args[@]}"; do
cmd="${cmd} ${arg}"
done
# 添加环境变量
for env in "${environment_vars[@]}"; do
cmd="${cmd} -e ${env}"
done
# 添加设备
for device in "${devices[@]}"; do
cmd="${cmd} --device=${device}"
done
# 添加挂载卷
for mount in "${mounts[@]}"; do
cmd="${cmd} -v ${mount}"
done
cmd="${cmd} ${docker_image} /bin/bash"
echo "${cmd}"
}
......@@ -85,7 +83,7 @@ fi
check_command_result() {
local exit_status=$1
if [ "$exit_status" -eq 0 ]; then
echo "命令成功执行"
else
......@@ -146,7 +144,7 @@ if [ "$num" -gt 0 ]; then
exit
fi
fi
# 如果是 -v 模式,显示信息后退出
if [ "$show_connect_cmd" = true ]; then
# 确保容器在运行状态
......@@ -160,12 +158,12 @@ if [ "$num" -gt 0 ]; then
show_container_info
exit
fi
# 默认行为:进入容器
docker start ${CONTAINER_NAME}
docker start ${CONTAINER_NAME}
docker exec -ti --env COLUMNS=`tput cols` --env LINES=`tput lines` ${CONTAINER_NAME} /bin/bash
exit
else
echo "Container:[${CONTAINER_NAME}] does not exist, will be created and go inside it."
......@@ -178,13 +176,13 @@ else
check_command_result $exit_status
echo "Container:[${CONTAINER_NAME}] Started successfully!"
# 如果是 -v 模式,显示信息后退出
if [ "$show_connect_cmd" = true ]; then
show_container_info
exit
fi
docker exec -ti --env COLUMNS=`tput cols` --env LINES=`tput lines` ${CONTAINER_NAME} /bin/bash
exit
fi
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment