text-gen-python-ampere-base (sha256:8685b86009902e2fbffa9e4c806e778879e4e00767ecf59ea311ebf233c7af2f)

Published 2024-05-21 05:08:17 -04:00 by simcop2387

Pull the image from the command line:

docker pull gitea.simcop2387.info/simcop2387/text-gen-python-ampere-base@sha256:8685b86009902e2fbffa9e4c806e778879e4e00767ecf59ea311ebf233c7af2f

Digest

sha256:8685b86009902e2fbffa9e4c806e778879e4e00767ecf59ea311ebf233c7af2f

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

ADD file:7f9a3c5a4231ed19174c21d17ce05d84d568cff6d3a0c2a1d7c3a9be5e45c02c in /

CMD ["/bin/bash"]

ENV NVARCH=x86_64

ENV NVIDIA_REQUIRE_CUDA=cuda>=12.3 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.3.101-1

ENV NV_CUDA_COMPAT_PACKAGE=cuda-compat-12-3

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 -fsSLO https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/${NVARCH}/cuda-keyring_1.0-1_all.deb && dpkg -i cuda-keyring_1.0-1_all.deb && apt-get purge --autoremove -y curl && rm -rf /var/lib/apt/lists/* # buildkit

ENV CUDA_VERSION=12.3.2

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

ENV NV_NVTX_VERSION=12.3.101-1

ENV NV_LIBNPP_VERSION=12.2.3.2-1

ENV NV_LIBNPP_PACKAGE=libnpp-12-3=12.2.3.2-1

ENV NV_LIBCUSPARSE_VERSION=12.2.0.103-1

ENV NV_LIBCUBLAS_PACKAGE_NAME=libcublas-12-3

ENV NV_LIBCUBLAS_VERSION=12.3.4.1-1

ENV NV_LIBCUBLAS_PACKAGE=libcublas-12-3=12.3.4.1-1

ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2

ENV NV_LIBNCCL_PACKAGE_VERSION=2.20.3-1

ENV NCCL_VERSION=2.20.3-1

ENV NV_LIBNCCL_PACKAGE=libnccl2=2.20.3-1+cuda12.3

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

ENV NV_CUDA_CUDART_DEV_VERSION=12.3.101-1

ENV NV_NVML_DEV_VERSION=12.3.101-1

ENV NV_LIBCUSPARSE_DEV_VERSION=12.2.0.103-1

ENV NV_LIBNPP_DEV_VERSION=12.2.3.2-1

ENV NV_LIBNPP_DEV_PACKAGE=libnpp-dev-12-3=12.2.3.2-1

ENV NV_LIBCUBLAS_DEV_VERSION=12.3.4.1-1

ENV NV_LIBCUBLAS_DEV_PACKAGE_NAME=libcublas-dev-12-3

ENV NV_LIBCUBLAS_DEV_PACKAGE=libcublas-dev-12-3=12.3.4.1-1

ENV NV_CUDA_NSIGHT_COMPUTE_VERSION=12.3.2-1

ENV NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE=cuda-nsight-compute-12-3=12.3.2-1

ENV NV_NVPROF_VERSION=12.3.101-1

ENV NV_NVPROF_DEV_PACKAGE=cuda-nvprof-12-3=12.3.101-1

ENV NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-dev

ENV NV_LIBNCCL_DEV_PACKAGE_VERSION=2.20.3-1

ENV NCCL_VERSION=2.20.3-1

ENV NV_LIBNCCL_DEV_PACKAGE=libnccl-dev=2.20.3-1+cuda12.3

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

ENV NV_CUDNN_PACKAGE_NAME=libcudnn9-cuda-12

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

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

ARG TARGETARCH

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

LABEL com.nvidia.cudnn.version=9.0.0.312

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 && apt install --no-install-recommends -y git vim build-essential python3-dev python3-venv && apt full-upgrade -y && rm -rf /var/lib/apt/lists/* # buildkit

ARG TORCH_CUDA_ARCH_LIST=3.5;5.0;6.0;6.1;7.0;7.5;8.0;8.6+PTX

LABEL maintainer=Your Name <your.email@example.com>

LABEL description=Docker image for GPTQ-for-LLaMa and Text Generation WebUI

RUN |1 TORCH_CUDA_ARCH_LIST=6.0;6.1;8.0;8.6+PTX /bin/sh -c apt-get update && apt-get install --no-install-recommends -y python3-dev libportaudio2 libasound-dev git python3 python3-pip make g++ ffmpeg && rm -rf /var/lib/apt/lists/* # buildkit

RUN |1 TORCH_CUDA_ARCH_LIST=6.0;6.1;8.0;8.6+PTX /bin/sh -c pip3 install virtualenv # buildkit

RUN |1 TORCH_CUDA_ARCH_LIST=6.0;6.1;8.0;8.6+PTX /bin/sh -c mkdir /app # buildkit

WORKDIR /app

ADD https://www.random.org/cgi-bin/randbyte?nbytes=10&format=h /tmp/skipcache # buildkit

RUN |1 TORCH_CUDA_ARCH_LIST=6.0;6.1;8.0;8.6+PTX /bin/sh -c git clone --recurse-submodules https://gitea.simcop2387.info/simcop2387/text-generation-webui-mirror . # buildkit

ARG WEBUI_VERSION

RUN |2 TORCH_CUDA_ARCH_LIST=6.0;6.1;8.0;8.6+PTX WEBUI_VERSION= /bin/sh -c test -n "${WEBUI_VERSION}" && git reset --hard ${WEBUI_VERSION} || echo "Using provided webui source" # buildkit

RUN |2 TORCH_CUDA_ARCH_LIST=6.0;6.1;8.0;8.6+PTX WEBUI_VERSION= /bin/sh -c virtualenv /app/venv # buildkit

RUN |2 TORCH_CUDA_ARCH_LIST=6.0;6.1;8.0;8.6+PTX WEBUI_VERSION= /bin/sh -c . /app/venv/bin/activate && pip3 install --upgrade pip setuptools wheel && pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 # buildkit

RUN |2 TORCH_CUDA_ARCH_LIST=6.0;6.1;8.0;8.6+PTX WEBUI_VERSION= /bin/sh -c . /app/venv/bin/activate && pip3 install -r requirements.txt # buildkit

COPY python-base/build-exts.sh /build-exts.sh # buildkit

RUN |2 TORCH_CUDA_ARCH_LIST=6.0;6.1;8.0;8.6+PTX WEBUI_VERSION= /bin/sh -c /build-exts.sh # buildkit

RUN |2 TORCH_CUDA_ARCH_LIST=6.0;6.1;8.0;8.6+PTX WEBUI_VERSION= /bin/sh -c cp /app/venv/lib/python3.10/site-packages/bitsandbytes/libbitsandbytes_cuda121.so /app/venv/lib/python3.10/site-packages/bitsandbytes/libbitsandbytes_cpu.so # buildkit

RUN |2 TORCH_CUDA_ARCH_LIST=6.0;6.1;8.0;8.6+PTX WEBUI_VERSION= /bin/sh -c . /app/venv/bin/activate && python3 -m nltk.downloader all # buildkit

ENV OPENEDAI_EMBEDDING_MODEL=jinaai/jina-embeddings-v2-base-en

ENV CLI_ARGS=

CMD ["/bin/sh" "-c" ". /app/venv/bin/activate && python3 server.py ${CLI_ARGS}"]

Key	Value
com.nvidia.cudnn.version	9.0.0.312
description	Docker image for GPTQ-for-LLaMa and Text Generation WebUI
maintainer	Your Name <your.email@example.com>
org.opencontainers.image.created	2024-05-21T08:36:17Z
org.opencontainers.image.ref.name	ubuntu
org.opencontainers.image.revision	45d597a641c01f71e627700d118e5bc3043e8c45
org.opencontainers.image.source	https://gitea.simcop2387.info/simcop2387/text-gen-base.git
org.opencontainers.image.url	https://gitea.simcop2387.info/simcop2387/text-gen-base
org.opencontainers.image.version	22.04

Details

Container

2024-05-21 05:08:17 -04:00

OCI / Docker

linux/amd64

Project website

Repository website

9.7 GiB

Versions (19) View all

build-38

2024-05-30

latest

2024-05-30

build-36

2024-05-23

build-35

2024-05-21

build-33

2024-05-20

text-gen-python-ampere-base (sha256:8685b86009902e2fbffa9e4c806e778879e4e00767ecf59ea311ebf233c7af2f)

Installation

Image layers

Labels