|
FROM pytorch/pytorch:2.4.0-cuda12.1-cudnn9-runtime |
|
|
|
ENV NB_USER jovyan |
|
ENV NB_UID 1000 |
|
ENV NB_PREFIX / |
|
|
|
RUN export DEBIAN_FRONTEND=noninteractive \ |
|
&& apt-get -yq update \ |
|
&& apt-get -yq install --no-install-recommends \ |
|
git \ |
|
ffmpeg \ |
|
software-properties-common \ |
|
&& apt upgrade -y \ |
|
&& apt-get clean \ |
|
&& rm -rf /var/lib/apt/lists/* |
|
|
|
RUN export DEBIAN_FRONTEND=noninteractive \ |
|
&& add-apt-repository ppa:deadsnakes/ppa \ |
|
&& apt-get -yq update \ |
|
&& apt-get -yq install --no-install-recommends \ |
|
python3.11 \ |
|
python3.11-venv \ |
|
&& apt upgrade -y \ |
|
&& apt-get clean \ |
|
&& rm -rf /var/lib/apt/lists/* |
|
|
|
RUN useradd -M -s /bin/bash -N -u ${NB_UID} ${NB_USER} \ |
|
&& mkdir -p ${HOME} \ |
|
&& chown -R ${NB_USER}:users ${HOME} \ |
|
&& chown -R ${NB_USER}:users /usr/local/bin |
|
|
|
USER $NB_UID |
|
WORKDIR /home/${NB_USER}/ |
|
ENV PATH=/venv/bin:$PATH |
|
|
|
COPY --chown=${NB_USER}:users ./requirements.txt /home/${NB_USER}/requirements.txt |
|
RUN python3.11 -m venv /home/${NB_USER}/venv \ |
|
&& /home/${NB_USER}/venv/bin/pip install --upgrade pip wheel \ |
|
&& /home/${NB_USER}/venv/bin/pip install -r /home/${NB_USER}/requirements.txt |
|
|
|
COPY --chown=${NB_USER}:users ./.env* /home/${NB_USER}/ |
|
COPY --chown=${NB_USER}:users ./src /home/${NB_USER}/src |
|
|
|
EXPOSE 7860 |
|
|
|
CMD ["/home/jovyan/venv/bin/python", "-u", "/home/jovyan/src/app.py"] |