如何通过Dockerfile的ENTRYPOINT运行脚本实现W&B登录？

1. 编写带W&B登录的Shell脚本 #

先创建一个名为run_with_wb_login.sh的脚本，它会先完成W&B登录，再执行后续业务命令：

#!/bin/bash
# 通过环境变量传入W&B API密钥完成登录
wandb login "$WANDB_API_KEY"

# 执行容器启动时指定的后续命令（用exec让命令成为PID 1，确保信号正常传递）
exec "$@"

2. 编写Dockerfile #

将脚本集成到Docker镜像中，配置ENTRYPOINT来启动这个脚本：

# 选择适配你业务的基础镜像，这里以Python环境为例
FROM python:3.10-slim

# 安装W&B CLI工具
RUN pip install --no-cache-dir wandb

# 设置工作目录
WORKDIR /app

# 复制本地的Shell脚本到容器内
COPY run_with_wb_login.sh .

# 给脚本添加执行权限
RUN chmod +x run_with_wb_login.sh

# 配置ENTRYPOINT，指定启动脚本
ENTRYPOINT ["./run_with_wb_login.sh"]

# 可选：设置默认的业务命令，比如你的训练脚本
CMD ["python", "train.py"]

3. 构建并运行容器 #

• 构建镜像：

docker build -t wb-training-image .

• 运行容器时通过环境变量传入W&B API密钥：

docker run -e WANDB_API_KEY=your_actual_api_key_here wb-training-image

如果需要覆盖默认CMD，直接在命令末尾追加业务命令即可：

docker run -e WANDB_API_KEY=your_actual_api_key_here wb-training-image python train.py --batch-size 32

关键注意事项 #

• 安全优先：绝对不要将W&B API密钥硬编码在脚本或Dockerfile中，始终用环境变量传递
• 信号处理：脚本中的exec "$@"会让后续业务命令成为容器的PID 1，确保Docker发送的终止信号（如docker stop）能被正确处理，避免容器强制退出
• 镜像适配：如果你的业务不需要Python环境，替换成对应的基础镜像（如Ubuntu）并安装W&B CLI即可

内容的提问来源于stack exchange，提问作者SJin