第一步,代码编写
Dockerfile
Dockerfile 的后半部分(设置环境变量默认值,以及传到 run.sh 中)
一个坑:ENTRYPOINT 后面不能用 [] 的形式 否则无法读取环境变量值。
# 设置环境默认值
ENV task_id=123
ENV task_param=1
# entrypoint是容器入口
# 不能用下面[]这种形式,否则读取不到环境变量!!
# ENTRYPOINT ["/bin/bash", "/code/run.sh"]
ENTRYPOINT /bin/bash /code/run.sh $task_id $task_param
run.sh
用来输出传过来的环境变量,用来验证环境变量是否生效;并执行 python 脚本
echo "task_id = ${task_id}"
echo "task_param = ${task_param}"
python /code/main.py
main.py
python 脚本,用os.environ
读取环境变量。
import os
task_id = os.environ['task_id']
task_param = os.environ['task_param']
第二步,运行 docker
构建镜像:docker build -t image_name:v1.211020 -f ./Dockerfile .
生成容器并运行:docker run -e task_id="ua084d" -e task_param={"tag": "BERT"} --name MO_CLS_ua084d 9c6ad6da --rm
这里踩的一个坑:
注意:docker run 的 - e 参数要紧跟 run,否则不生效!!
注意:docker run 的 - e 参数要紧跟 run,否则不生效!!
注意:docker run 的 - e 参数要紧跟 run,否则不生效!!
转载请注明出处