vllm-custom-image (sha256:55fecb61bbebda60c273d2202ae1d4ee1bb881bb8b2f5463c48343db057ddd6d)

Published 2024-10-30 04:59:36 -04:00 by simcop2387

Pull the image from the command line:

docker pull gitea.simcop2387.info/simcop2387/vllm-custom-image@sha256:55fecb61bbebda60c273d2202ae1d4ee1bb881bb8b2f5463c48343db057ddd6d

Digest

sha256:55fecb61bbebda60c273d2202ae1d4ee1bb881bb8b2f5463c48343db057ddd6d

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=20.04

ADD file:ea2128e23dce0162557abadd80656bd5ae047d573095d1d4323eb4154490dfdc in /

CMD ["/bin/bash"]

ENV NVARCH=x86_64

ENV NVIDIA_REQUIRE_CUDA=cuda>=12.4 brand=tesla,driver>=470,driver<471 brand=unknown,driver>=470,driver<471 brand=nvidia,driver>=470,driver<471 brand=nvidiartx,driver>=470,driver<471 brand=geforce,driver>=470,driver<471 brand=geforcertx,driver>=470,driver<471 brand=quadro,driver>=470,driver<471 brand=quadrortx,driver>=470,driver<471 brand=titan,driver>=470,driver<471 brand=titanrtx,driver>=470,driver<471 brand=tesla,driver>=525,driver<526 brand=unknown,driver>=525,driver<526 brand=nvidia,driver>=525,driver<526 brand=nvidiartx,driver>=525,driver<526 brand=geforce,driver>=525,driver<526 brand=geforcertx,driver>=525,driver<526 brand=quadro,driver>=525,driver<526 brand=quadrortx,driver>=525,driver<526 brand=titan,driver>=525,driver<526 brand=titanrtx,driver>=525,driver<526 brand=tesla,driver>=535,driver<536 brand=unknown,driver>=535,driver<536 brand=nvidia,driver>=535,driver<536 brand=nvidiartx,driver>=535,driver<536 brand=geforce,driver>=535,driver<536 brand=geforcertx,driver>=535,driver<536 brand=quadro,driver>=535,driver<536 brand=quadrortx,driver>=535,driver<536 brand=titan,driver>=535,driver<536 brand=titanrtx,driver>=535,driver<536

ENV NV_CUDA_CUDART_VERSION=12.4.127-1

ENV NV_CUDA_COMPAT_PACKAGE=cuda-compat-12-4

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/ubuntu2004/${NVARCH}/3bf863cc.pub | apt-key add - && echo "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/${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.4.1

RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends cuda-cudart-12-4=${NV_CUDA_CUDART_VERSION} ${NV_CUDA_COMPAT_PACKAGE} && 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.4.1-1

ENV NV_NVTX_VERSION=12.4.127-1

ENV NV_LIBNPP_VERSION=12.2.5.30-1

ENV NV_LIBNPP_PACKAGE=libnpp-12-4=12.2.5.30-1

ENV NV_LIBCUSPARSE_VERSION=12.3.1.170-1

ENV NV_LIBCUBLAS_PACKAGE_NAME=libcublas-12-4

ENV NV_LIBCUBLAS_VERSION=12.4.5.8-1

ENV NV_LIBCUBLAS_PACKAGE=libcublas-12-4=12.4.5.8-1

ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2

ENV NV_LIBNCCL_PACKAGE_VERSION=2.21.5-1

ENV NCCL_VERSION=2.21.5-1

ENV NV_LIBNCCL_PACKAGE=libnccl2=2.21.5-1+cuda12.4

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-4=${NV_CUDA_LIB_VERSION} ${NV_LIBNPP_PACKAGE} cuda-nvtx-12-4=${NV_NVTX_VERSION} libcusparse-12-4=${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.4.1-1

ENV NV_CUDA_CUDART_DEV_VERSION=12.4.127-1

ENV NV_NVML_DEV_VERSION=12.4.127-1

ENV NV_LIBCUSPARSE_DEV_VERSION=12.3.1.170-1

ENV NV_LIBNPP_DEV_VERSION=12.2.5.30-1

ENV NV_LIBNPP_DEV_PACKAGE=libnpp-dev-12-4=12.2.5.30-1

ENV NV_LIBCUBLAS_DEV_VERSION=12.4.5.8-1

ENV NV_LIBCUBLAS_DEV_PACKAGE_NAME=libcublas-dev-12-4

ENV NV_LIBCUBLAS_DEV_PACKAGE=libcublas-dev-12-4=12.4.5.8-1

ENV NV_CUDA_NSIGHT_COMPUTE_VERSION=12.4.1-1

ENV NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE=cuda-nsight-compute-12-4=12.4.1-1

ENV NV_NVPROF_VERSION=12.4.127-1

ENV NV_NVPROF_DEV_PACKAGE=cuda-nvprof-12-4=12.4.127-1

ENV NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-dev

ENV NV_LIBNCCL_DEV_PACKAGE_VERSION=2.21.5-1

ENV NCCL_VERSION=2.21.5-1

ENV NV_LIBNCCL_DEV_PACKAGE=libnccl-dev=2.21.5-1+cuda12.4

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 libtinfo5 libncursesw5 cuda-cudart-dev-12-4=${NV_CUDA_CUDART_DEV_VERSION} cuda-command-line-tools-12-4=${NV_CUDA_LIB_VERSION} cuda-minimal-build-12-4=${NV_CUDA_LIB_VERSION} cuda-libraries-dev-12-4=${NV_CUDA_LIB_VERSION} cuda-nvml-dev-12-4=${NV_NVML_DEV_VERSION} ${NV_NVPROF_DEV_PACKAGE} ${NV_LIBNPP_DEV_PACKAGE} libcusparse-dev-12-4=${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

ARG CUDA_VERSION=12.4.1

ARG PYTHON_VERSION=3.11

WORKDIR /vllm-workspace

ENV DEBIAN_FRONTEND=noninteractive

RUN |2 CUDA_VERSION=12.4.1 PYTHON_VERSION=3.11 /bin/sh -c PYTHON_VERSION_STR=$(echo ${PYTHON_VERSION} | sed 's/\.//g') && echo "export PYTHON_VERSION_STR=${PYTHON_VERSION_STR}" >> /etc/environment # buildkit

RUN |2 CUDA_VERSION=12.4.1 PYTHON_VERSION=3.11 /bin/sh -c echo 'tzdata tzdata/Areas select America' | debconf-set-selections && echo 'tzdata tzdata/Zones/America select Los_Angeles' | debconf-set-selections && apt-get update -y && apt-get install -y ccache software-properties-common git curl sudo vim python3-pip && apt-get install -y ffmpeg libsm6 libxext6 libgl1 && add-apt-repository ppa:deadsnakes/ppa && apt-get update -y && apt-get install -y python${PYTHON_VERSION} python${PYTHON_VERSION}-dev python${PYTHON_VERSION}-venv libibverbs-dev && update-alternatives --install /usr/bin/python3 python3 /usr/bin/python${PYTHON_VERSION} 1 && update-alternatives --set python3 /usr/bin/python${PYTHON_VERSION} && ln -sf /usr/bin/python${PYTHON_VERSION}-config /usr/bin/python3-config && curl -sS https://bootstrap.pypa.io/get-pip.py | python${PYTHON_VERSION} && python3 --version && python3 -m pip --version # buildkit

RUN |2 CUDA_VERSION=12.4.1 PYTHON_VERSION=3.11 /bin/sh -c ldconfig /usr/local/cuda-$(echo $CUDA_VERSION | cut -d. -f1,2)/compat/ # buildkit

RUN |2 CUDA_VERSION=12.4.1 PYTHON_VERSION=3.11 /bin/sh -c python3 -m pip install dist/*.whl --verbose # buildkit

RUN |2 CUDA_VERSION=12.4.1 PYTHON_VERSION=3.11 /bin/sh -c . /etc/environment && python3 -m pip install https://github.com/flashinfer-ai/flashinfer/releases/download/v0.1.6/flashinfer-0.1.6+cu121torch2.4-cp${PYTHON_VERSION_STR}-cp${PYTHON_VERSION_STR}-linux_x86_64.whl && python3 -m pip install flash_attn llmcompressor mistral_common[opencv] # buildkit

RUN |2 CUDA_VERSION=12.4.1 PYTHON_VERSION=3.11 /bin/sh -c pip install accelerate hf_transfer 'modelscope!=1.15.0' bitsandbytes>=0.44.0 timm==0.9.10 # buildkit

ENV VLLM_USAGE_SOURCE=production-docker-image

ENTRYPOINT ["python3" "-m" "vllm.entrypoints.openai.api_server"]

Key	Value
maintainer	NVIDIA CORPORATION <cudatools@nvidia.com>
org.opencontainers.image.created	2024-10-30T01:30:31Z
org.opencontainers.image.ref.name	ubuntu
org.opencontainers.image.revision	26b03be5cbfb879ac277751fd0d48c86fba0b394
org.opencontainers.image.source	https://gitea.simcop2387.info/simcop2387/vllm-containers.git
org.opencontainers.image.url	https://gitea.simcop2387.info/simcop2387/vllm-containers
org.opencontainers.image.version	20.04

Details

Container

2024-10-30 04:59:36 -04:00

OCI / Docker

linux/amd64

Project website

Repository website

9 GiB

Versions (11) View all

latest

2024-10-30

build-77

2024-10-30

build-74

2024-10-28

build-73

2024-10-23

build-71

2024-10-14

vllm-custom-image (sha256:55fecb61bbebda60c273d2202ae1d4ee1bb881bb8b2f5463c48343db057ddd6d)

Installation

Image layers

Labels