fix(containerize): command to create group and user in alpine base im…

…age (#4855) Signed-off-by: Frost Ming <[email protected]>
bentoml · Jul 12, 2024 · f40a11f · f40a11f
1 parent 3154de1
commit f40a11f
Show file tree

Hide file tree

Showing 5 changed files with 20 additions and 1 deletion.
diff --git a/src/bentoml/_internal/container/frontend/dockerfile/templates/base.j2 b/src/bentoml/_internal/container/frontend/dockerfile/templates/base.j2
@@ -39,7 +39,6 @@ ENV NVIDIA_VISIBLE_DEVICES=
 ARG BENTO_USER={{ bento__user }}
 ARG BENTO_USER_UID={{ bento__uid_gid }}
 ARG BENTO_USER_GID={{ bento__uid_gid }}
-RUN groupadd -g $BENTO_USER_GID -o $BENTO_USER && useradd -m -u $BENTO_USER_UID -g $BENTO_USER_GID -o -r $BENTO_USER
 {% endblock %}
 {% block SETUP_BENTO_ENVARS %}
 {% if __options__env is not none %}

diff --git a/src/bentoml/_internal/container/frontend/dockerfile/templates/base_alpine.j2 b/src/bentoml/_internal/container/frontend/dockerfile/templates/base_alpine.j2
@@ -18,3 +18,8 @@ RUN curl -LO https://astral.sh/uv/install.sh && \
     apk add --update {{ __options__system_packages | join(' ') }}
 {% endif -%}
 {% endblock %}
+
+{% block SETUP_BENTO_USER %}
+{{ super() }}
+RUN addgroup -g $BENTO_USER_GID -S $BENTO_USER && adduser -u $BENTO_USER_UID -G $BENTO_USER -S $BENTO_USER
+{% endblock %}
diff --git a/src/bentoml/_internal/container/frontend/dockerfile/templates/base_amazonlinux.j2 b/src/bentoml/_internal/container/frontend/dockerfile/templates/base_amazonlinux.j2
@@ -24,3 +24,8 @@ RUN curl -LO https://astral.sh/uv/install.sh && \
 RUN ln -sf /usr/bin/python{{ __options__python_version }} /usr/bin/python3 && \
     ln -sf /usr/bin/pip{{ __options__python_version }} /usr/bin/pip3
 {% endblock %}
+
+{% block SETUP_BENTO_USER %}
+{{ super() }}
+RUN groupadd -g $BENTO_USER_GID -o $BENTO_USER && useradd -m -u $BENTO_USER_UID -g $BENTO_USER_GID -o -r $BENTO_USER
+{% endblock %}
diff --git a/src/bentoml/_internal/container/frontend/dockerfile/templates/base_debian.j2 b/src/bentoml/_internal/container/frontend/dockerfile/templates/base_debian.j2
@@ -16,3 +16,8 @@ ENV UV_SYSTEM_PYTHON=1
 RUN curl -LO https://astral.sh/uv/install.sh && \
     sh install.sh && rm install.sh && mv $HOME/.cargo/bin/uv /usr/local/bin/
 {% endblock %}
+
+{% block SETUP_BENTO_USER %}
+{{ super() }}
+RUN groupadd -g $BENTO_USER_GID -o $BENTO_USER && useradd -m -u $BENTO_USER_UID -g $BENTO_USER_GID -o -r $BENTO_USER
+{% endblock %}
diff --git a/src/bentoml/_internal/container/frontend/dockerfile/templates/base_ubi8.j2 b/src/bentoml/_internal/container/frontend/dockerfile/templates/base_ubi8.j2
@@ -16,3 +16,8 @@ RUN curl -LO https://astral.sh/uv/install.sh && \
     yum install -y {{ __options__system_packages | join(' ') }}
 {% endif -%}
 {% endblock %}
+
+{% block SETUP_BENTO_USER %}
+{{ super() }}
+RUN groupadd -g $BENTO_USER_GID -o $BENTO_USER && useradd -m -u $BENTO_USER_UID -g $BENTO_USER_GID -o -r $BENTO_USER
+{% endblock %}