docker_template/Dockerfile

FROM pytorch/pytorch:2.0.0-cuda11.7-cudnn8-runtime
#FROM tensorflow/tensorflow:2.12.0-gpu

ARG USER
ARG GROUP
ARG UID
ARG GID

RUN groupadd --gid ${GID} ${GROUP}
RUN useradd --shell /bin/bash --uid ${UID} --gid ${GID} --create-home ${USER}
RUN mkdir /wd
RUN chown ${USER}:${GROUP} /wd

# SYSTEM INITIALIZATION
# RUN pip install jupyter

USER ${USER}

# USER INITIALIZATION
# RUN ...

SHELL ["/bin/bash", "--login", "-i", "-c"]
WORKDIR /wd