base-agent (gpu-amd64-2024-12-21)

2024-12-21T19:30:08+01:00

Pull the image from the command line:

docker pull git.chai.uni-hamburg.de/ums-agenten/base-agent:gpu-amd64-2024-12-21

Digest

sha256:4a7efaa6073080b030a7fa5345cf68a080c2aa5bfdc2e65ae2f19ee6c963a8ee

For more information on the Container registry, see the documentation.

ARG RELEASE

ARG LAUNCHPAD_BUILD_ARCH

LABEL org.opencontainers.image.ref.name=ubuntu

LABEL org.opencontainers.image.version=24.04

ADD file:c2e78eb585ec4e503f14c4ea98f4962c998f5eb075749507953f85387742694b in /

CMD ["/bin/bash"]

ENV NVARCH=x86_64

ENV NVIDIA_REQUIRE_CUDA=cuda>=12.6 brand=unknown,driver>=470,driver<471 brand=grid,driver>=470,driver<471 brand=tesla,driver>=470,driver<471 brand=nvidia,driver>=470,driver<471 brand=quadro,driver>=470,driver<471 brand=quadrortx,driver>=470,driver<471 brand=nvidiartx,driver>=470,driver<471 brand=vapps,driver>=470,driver<471 brand=vpc,driver>=470,driver<471 brand=vcs,driver>=470,driver<471 brand=vws,driver>=470,driver<471 brand=cloudgaming,driver>=470,driver<471 brand=unknown,driver>=535,driver<536 brand=grid,driver>=535,driver<536 brand=tesla,driver>=535,driver<536 brand=nvidia,driver>=535,driver<536 brand=quadro,driver>=535,driver<536 brand=quadrortx,driver>=535,driver<536 brand=nvidiartx,driver>=535,driver<536 brand=vapps,driver>=535,driver<536 brand=vpc,driver>=535,driver<536 brand=vcs,driver>=535,driver<536 brand=vws,driver>=535,driver<536 brand=cloudgaming,driver>=535,driver<536 brand=unknown,driver>=550,driver<551 brand=grid,driver>=550,driver<551 brand=tesla,driver>=550,driver<551 brand=nvidia,driver>=550,driver<551 brand=quadro,driver>=550,driver<551 brand=quadrortx,driver>=550,driver<551 brand=nvidiartx,driver>=550,driver<551 brand=vapps,driver>=550,driver<551 brand=vpc,driver>=550,driver<551 brand=vcs,driver>=550,driver<551 brand=vws,driver>=550,driver<551 brand=cloudgaming,driver>=550,driver<551

ENV NV_CUDA_CUDART_VERSION=12.6.68-1

ARG TARGETARCH

LABEL maintainer=NVIDIA CORPORATION <cudatools@nvidia.com>

RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends gnupg2 curl ca-certificates && curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/${NVARCH}/3bf863cc.pub | apt-key add - && echo "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/${NVARCH} /" > /etc/apt/sources.list.d/cuda.list && apt-get purge --autoremove -y curl && rm -rf /var/lib/apt/lists/* # buildkit

ENV CUDA_VERSION=12.6.1

RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends cuda-cudart-12-6=${NV_CUDA_CUDART_VERSION} cuda-compat-12-6 && rm -rf /var/lib/apt/lists/* # buildkit

RUN |1 TARGETARCH=amd64 /bin/sh -c echo "/usr/local/nvidia/lib" >> /etc/ld.so.conf.d/nvidia.conf && echo "/usr/local/nvidia/lib64" >> /etc/ld.so.conf.d/nvidia.conf # buildkit

ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64

COPY NGC-DL-CONTAINER-LICENSE / # buildkit

ENV NVIDIA_VISIBLE_DEVICES=all

ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility

ENV NV_CUDA_LIB_VERSION=12.6.1-1

ENV NV_NVTX_VERSION=12.6.68-1

ENV NV_LIBNPP_VERSION=12.3.1.54-1

ENV NV_LIBNPP_PACKAGE=libnpp-12-6=12.3.1.54-1

ENV NV_LIBCUSPARSE_VERSION=12.5.3.3-1

ENV NV_LIBCUBLAS_PACKAGE_NAME=libcublas-12-6

ENV NV_LIBCUBLAS_VERSION=12.6.1.4-1

ENV NV_LIBCUBLAS_PACKAGE=libcublas-12-6=12.6.1.4-1

ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2

ENV NV_LIBNCCL_PACKAGE_VERSION=2.22.3-1

ENV NCCL_VERSION=2.22.3-1

ENV NV_LIBNCCL_PACKAGE=libnccl2=2.22.3-1+cuda12.6

ARG TARGETARCH

LABEL maintainer=NVIDIA CORPORATION <cudatools@nvidia.com>

RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends cuda-libraries-12-6=${NV_CUDA_LIB_VERSION} ${NV_LIBNPP_PACKAGE} cuda-nvtx-12-6=${NV_NVTX_VERSION} libcusparse-12-6=${NV_LIBCUSPARSE_VERSION} ${NV_LIBCUBLAS_PACKAGE} ${NV_LIBNCCL_PACKAGE} && rm -rf /var/lib/apt/lists/* # buildkit

RUN |1 TARGETARCH=amd64 /bin/sh -c apt-mark hold ${NV_LIBCUBLAS_PACKAGE_NAME} ${NV_LIBNCCL_PACKAGE_NAME} # buildkit

COPY entrypoint.d/ /opt/nvidia/entrypoint.d/ # buildkit

COPY nvidia_entrypoint.sh /opt/nvidia/ # buildkit

ENV NVIDIA_PRODUCT_NAME=CUDA

ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"]

ENV NV_CUDA_LIB_VERSION=12.6.1-1

ENV NV_CUDA_CUDART_DEV_VERSION=12.6.68-1

ENV NV_NVML_DEV_VERSION=12.6.68-1

ENV NV_LIBCUSPARSE_DEV_VERSION=12.5.3.3-1

ENV NV_LIBNPP_DEV_VERSION=12.3.1.54-1

ENV NV_LIBNPP_DEV_PACKAGE=libnpp-dev-12-6=12.3.1.54-1

ENV NV_LIBCUBLAS_DEV_VERSION=12.6.1.4-1

ENV NV_LIBCUBLAS_DEV_PACKAGE_NAME=libcublas-dev-12-6

ENV NV_LIBCUBLAS_DEV_PACKAGE=libcublas-dev-12-6=12.6.1.4-1

ENV NV_CUDA_NSIGHT_COMPUTE_VERSION=12.6.1-1

ENV NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE=cuda-nsight-compute-12-6=12.6.1-1

ENV NV_NVPROF_VERSION=12.6.68-1

ENV NV_NVPROF_DEV_PACKAGE=cuda-nvprof-12-6=12.6.68-1

ENV NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-dev

ENV NV_LIBNCCL_DEV_PACKAGE_VERSION=2.22.3-1

ENV NCCL_VERSION=2.22.3-1

ENV NV_LIBNCCL_DEV_PACKAGE=libnccl-dev=2.22.3-1+cuda12.6

ARG TARGETARCH

LABEL maintainer=NVIDIA CORPORATION <cudatools@nvidia.com>

RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends cuda-cudart-dev-12-6=${NV_CUDA_CUDART_DEV_VERSION} cuda-command-line-tools-12-6=${NV_CUDA_LIB_VERSION} cuda-minimal-build-12-6=${NV_CUDA_LIB_VERSION} cuda-libraries-dev-12-6=${NV_CUDA_LIB_VERSION} cuda-nvml-dev-12-6=${NV_NVML_DEV_VERSION} ${NV_NVPROF_DEV_PACKAGE} ${NV_LIBNPP_DEV_PACKAGE} libcusparse-dev-12-6=${NV_LIBCUSPARSE_DEV_VERSION} ${NV_LIBCUBLAS_DEV_PACKAGE} ${NV_LIBNCCL_DEV_PACKAGE} ${NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE} && rm -rf /var/lib/apt/lists/* # buildkit

RUN |1 TARGETARCH=amd64 /bin/sh -c apt-mark hold ${NV_LIBCUBLAS_DEV_PACKAGE_NAME} ${NV_LIBNCCL_DEV_PACKAGE_NAME} # buildkit

ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs

ENV NV_CUDNN_VERSION=9.3.0.75-1

ENV NV_CUDNN_PACKAGE_NAME=libcudnn9-cuda-12

ENV NV_CUDNN_PACKAGE=libcudnn9-cuda-12=9.3.0.75-1

ENV NV_CUDNN_PACKAGE_DEV=libcudnn9-dev-cuda-12=9.3.0.75-1

ARG TARGETARCH

LABEL maintainer=NVIDIA CORPORATION <cudatools@nvidia.com>

LABEL com.nvidia.cudnn.version=9.3.0.75-1

RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends ${NV_CUDNN_PACKAGE} ${NV_CUDNN_PACKAGE_DEV} && apt-mark hold ${NV_CUDNN_PACKAGE_NAME} && rm -rf /var/lib/apt/lists/* # buildkit

RUN /bin/sh -c apt-get update && DEBIAN_FRONTEND=noninteractive TZ=Europe/Berlin apt-get install -y tzdata && cp /usr/share/zoneinfo/Europe/Berlin /etc/localtime && echo "Europe/Berlin" > /etc/timezone # buildkit

RUN /bin/sh -c apt update && apt install -y bash build-essential git curl ca-certificates python3-dev python3-pip libopenmpi-dev && rm -rf /var/lib/apt/lists # buildkit

RUN /bin/sh -c python3 -m pip install --no-cache-dir --break-system-packages torch --index-url https://download.pytorch.org/whl/cu124 # buildkit

ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/cuda/lib64

RUN /bin/sh -c ln -s /usr/local/cuda/lib64/libcusolver.so.11 /usr/local/cuda/lib64/libcusolver.so.10 # buildkit

CMD ["/bin/bash"]

ARG H_GID=1050

ARG H_UID=1050

ARG PIP_REQ_FILE=requirements-frozen.txt

RUN |3 H_GID=1050 H_UID=1050 PIP_REQ_FILE=requirements-frozen.txt /bin/sh -c apt update && DEBIAN_FRONTEND=noninteractive TZ=Europe/Berlin apt-get install -y tzdata && cp /usr/share/zoneinfo/Europe/Berlin /etc/localtime && echo "Europe/Berlin" > /etc/timezone # buildkit

RUN |3 H_GID=1050 H_UID=1050 PIP_REQ_FILE=requirements-frozen.txt /bin/sh -c apt install -y bash build-essential git curl ca-certificates python3-dev python3-pip # buildkit

RUN |3 H_GID=1050 H_UID=1050 PIP_REQ_FILE=requirements-frozen.txt /bin/sh -c ln -s /usr/bin/python3 /usr/local/bin/python && addgroup --gid $H_GID user && adduser user --uid $H_UID --ingroup user --gecos "" --home /home/user/ --disabled-password # buildkit

RUN |3 H_GID=1050 H_UID=1050 PIP_REQ_FILE=requirements-frozen.txt /bin/sh -c apt-get update && apt-get install -y bash nano vim mc curl wget htop screen # buildkit

RUN |3 H_GID=1050 H_UID=1050 PIP_REQ_FILE=requirements-frozen.txt /bin/sh -c if [ ! -d /usr/local/cuda/ ]; then python3 -m pip install --no-cache-dir --break-system-packages torch --index-url https://download.pytorch.org/whl/cpu; python3 -m pip install --no-cache-dir --break-system-packages faiss-cpu; else python3 -m pip install --no-cache-dir --break-system-packages faiss-gpu-cu12; fi; # buildkit

RUN |3 H_GID=1050 H_UID=1050 PIP_REQ_FILE=requirements-frozen.txt /bin/sh -c mkdir -p /ums-agenten/persist/ # buildkit

ENV NLTK_DATA=/ums-agenten/persist/nltk/

ENV HF_HOME=/ums-agenten/persist/hf_home/

COPY ./docker/requirements-frozen.txt /ums-agenten/requirements.txt # buildkit

RUN |3 H_GID=1050 H_UID=1050 PIP_REQ_FILE=requirements-frozen.txt /bin/sh -c pip3 install --break-system-packages --no-cache-dir -r /ums-agenten/requirements.txt && python -m pip freeze > /ums-agenten/requirements.txt # buildkit

COPY ./docker/init.py /ums-agenten/project/ # buildkit

COPY ./docker/setup.py /ums-agenten/project/ # buildkit

RUN |3 H_GID=1050 H_UID=1050 PIP_REQ_FILE=requirements-frozen.txt /bin/sh -c pip3 install --break-system-packages -e /ums-agenten/project/ # buildkit

WORKDIR /ums-agenten/project/src/

RUN |3 H_GID=1050 H_UID=1050 PIP_REQ_FILE=requirements-frozen.txt /bin/sh -c chown -R user:user /ums-agenten # buildkit

USER user

CMD ["/bin/bash"]

ARG PIP_REQ_FILE=requirements-frozen.txt

USER root

RUN |1 PIP_REQ_FILE=requirements-frozen.txt /bin/sh -c mkdir -p /ums-agenten/plattform/ && mkdir -p /ums-agenten/persist/ # buildkit

COPY ./utils/agent/requirements-frozen.txt /ums-agenten/requirements.txt # buildkit

RUN |1 PIP_REQ_FILE=requirements-frozen.txt /bin/sh -c pip3 install --break-system-packages --no-cache-dir -r /ums-agenten/requirements.txt && pip3 freeze -q -r /ums-agenten/requirements.txt > /ums-agenten/requirements-frozen.txt # buildkit

COPY ./utils/setup.py /ums-agenten/plattform/ # buildkit

RUN |1 PIP_REQ_FILE=requirements-frozen.txt /bin/sh -c pip3 install --break-system-packages -e /ums-agenten/plattform/ # buildkit

COPY --chown=user:user ./ums/ /ums-agenten/plattform/ums/ # buildkit

COPY --chown=user:user ./web/ /ums-agenten/plattform/web/ # buildkit

WORKDIR /ums-agenten/plattform/ums/

RUN |1 PIP_REQ_FILE=requirements-frozen.txt /bin/sh -c chown -R user:user /ums-agenten # buildkit

USER user

ENV SERVE=true

CMD ["/usr/local/bin/uvicorn" "ums.agent.main:app" "--host" "0.0.0.0" "--port" "8000"]

Key	Value
com.nvidia.cudnn.version	9.3.0.75-1
maintainer	NVIDIA CORPORATION <cudatools@nvidia.com>
org.opencontainers.image.ref.name	ubuntu
org.opencontainers.image.version	24.04