compreface-core (1.2.0-arcface-r100-gpu)

Published 2025-01-26 14:49:37 +01:00 by naamval

Pull the image from the command line:

docker pull forgejo.naamval.net/naamval/compreface-core:1.2.0-arcface-r100-gpu

Digest

sha256:af2fba33a6b6cb50da327cceb7f68f79431db7e99a603d475900af3ba775f019

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:655d373cb551d0dd5d7867f88a4f98908dc3f16190986f693e88c423e6f21b8d in /

CMD ["/bin/bash"]

ENV NVARCH=x86_64

ENV NVIDIA_REQUIRE_CUDA=cuda>=11.8 brand=tesla,driver>=450,driver<451 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>=510,driver<511 brand=unknown,driver>=510,driver<511 brand=nvidia,driver>=510,driver<511 brand=nvidiartx,driver>=510,driver<511 brand=geforce,driver>=510,driver<511 brand=geforcertx,driver>=510,driver<511 brand=quadro,driver>=510,driver<511 brand=quadrortx,driver>=510,driver<511 brand=titan,driver>=510,driver<511 brand=titanrtx,driver>=510,driver<511 brand=tesla,driver>=515,driver<516 brand=unknown,driver>=515,driver<516 brand=nvidia,driver>=515,driver<516 brand=nvidiartx,driver>=515,driver<516 brand=geforce,driver>=515,driver<516 brand=geforcertx,driver>=515,driver<516 brand=quadro,driver>=515,driver<516 brand=quadrortx,driver>=515,driver<516 brand=titan,driver>=515,driver<516 brand=titanrtx,driver>=515,driver<516

ENV NV_CUDA_CUDART_VERSION=11.8.89-1

ENV NV_CUDA_COMPAT_PACKAGE=cuda-compat-11-8

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

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

ENV NV_NVTX_VERSION=11.8.86-1

ENV NV_LIBNPP_VERSION=11.8.0.86-1

ENV NV_LIBNPP_PACKAGE=libnpp-11-8=11.8.0.86-1

ENV NV_LIBCUSPARSE_VERSION=11.7.5.86-1

ENV NV_LIBCUBLAS_PACKAGE_NAME=libcublas-11-8

ENV NV_LIBCUBLAS_VERSION=11.11.3.6-1

ENV NV_LIBCUBLAS_PACKAGE=libcublas-11-8=11.11.3.6-1

ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2

ENV NV_LIBNCCL_PACKAGE_VERSION=2.16.2-1

ENV NCCL_VERSION=2.16.2-1

ENV NV_LIBNCCL_PACKAGE=libnccl2=2.16.2-1+cuda11.8

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-11-8=${NV_CUDA_LIB_VERSION} ${NV_LIBNPP_PACKAGE} cuda-nvtx-11-8=${NV_NVTX_VERSION} libcusparse-11-8=${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_CUDNN_VERSION=8.9.0.131

ENV NV_CUDNN_PACKAGE_NAME=libcudnn8

ENV NV_CUDNN_PACKAGE=libcudnn8=8.9.0.131-1+cuda11.8

ARG TARGETARCH

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

LABEL com.nvidia.cudnn.version=8.9.0.131

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

ENV DEBIAN_FRONTEND=noninteractive

ENV CUDA=11.8

RUN /bin/sh -c apt-get update && apt-get install -y --no-install-recommends build-essential software-properties-common curl pkg-config unzip python3-dev python3-distutils && rm -rf /var/lib/apt/lists/* # buildkit

ENV LANG=C.UTF-8

RUN /bin/sh -c curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py && python3 get-pip.py # buildkit

RUN /bin/sh -c python3 -m pip --no-cache-dir install --upgrade pip setuptools # buildkit

RUN /bin/sh -c ln -s $(which python3) /usr/local/bin/python # buildkit

ENV TF_FORCE_GPU_ALLOW_GROWTH=true

ENV MXNET_CPU_WORKER_NTHREADS=24

ENV MXNET_ENGINE_TYPE=ThreadedEnginePerDevice MXNET_CUDNN_AUTOTUNE_DEFAULT=0

ENV SKIP_TESTS=1

RUN /bin/sh -c apt-get update && apt-get install -y build-essential cmake git wget unzip curl yasm pkg-config libswscale-dev libtbb2 libtbb-dev libjpeg-dev libpng-dev libtiff-dev libavformat-dev libpq-dev libfreeimage3 python3-opencv libaec-dev libblosc-dev libbrotli-dev libbz2-dev libgif-dev libopenjp2-7-dev liblcms2-dev libcharls-dev libjxr-dev liblz4-dev libcfitsio-dev libpcre3 libpcre3-dev libsnappy-dev libwebp-dev libzopfli-dev libzstd-dev && rm -rf /var/lib/apt/lists/* # buildkit

WORKDIR /tmp

RUN /bin/sh -c git clone --depth=1 --shallow-submodules --recursive -b v0.1 https://github.com/google/brunsli && cd brunsli && cmake -DCMAKE_BUILD_TYPE=Release . && make -j$(nproc) install && rm -rf /tmp/brunsli # buildkit

RUN /bin/sh -c git clone --depth=1 --shallow-submodules --recursive -b v0.7.0 https://github.com/libjxl/libjxl && cd libjxl && cmake -DCMAKE_BUILD_TYPE=Release -DBUILD_TESTING=OFF . && make -j$(nproc) install && rm -rf /tmp/libjxl # buildkit

RUN /bin/sh -c git clone --depth=1 -b 0.5.5 https://github.com/LLNL/zfp && cd zfp && mkdir build && cd build && cmake -DCMAKE_BUILD_TYPE=Release .. && make -j$(nproc) install && rm -rf /tmp/zfp # buildkit

SHELL [/bin/bash -c]

WORKDIR /app/ml

COPY requirements.txt . # buildkit

RUN /bin/bash -c pip --no-cache-dir install $(grep ^numpy requirements.txt) # buildkit

RUN /bin/bash -c pip --no-cache-dir install -r requirements.txt # buildkit

ARG BE_VERSION

ARG APP_VERSION_STRING

ENV BE_VERSION=

ENV APP_VERSION_STRING=

ENV HOME=/app/ml

ENV LC_ALL=C.UTF-8

ENV LANG=C.UTF-8

ENV PYTHONUNBUFFERED=0

ENV JOBLIB_MULTIPROCESSING=0

ARG INTEL_OPTIMIZATION=false

ARG GPU_IDX=-1

ENV GPU_IDX=0 INTEL_OPTIMIZATION=false

ARG FACE_DETECTION_PLUGIN=facenet.FaceDetector

ARG CALCULATION_PLUGIN=facenet.Calculator

ARG EXTRA_PLUGINS=facenet.LandmarksDetector,agegender.AgeDetector,agegender.GenderDetector,facenet.facemask.MaskDetector,facenet.PoseEstimator

ENV FACE_DETECTION_PLUGIN=insightface.FaceDetector@retinaface_r50_v1 CALCULATION_PLUGIN=insightface.Calculator@arcface-r100-msfdrop75 EXTRA_PLUGINS=insightface.LandmarksDetector,insightface.GenderDetector,insightface.AgeDetector,insightface.facemask.MaskDetector,insightface.PoseEstimator

COPY src src # buildkit

COPY srcext srcext # buildkit

RUN |7 BE_VERSION= APP_VERSION_STRING= INTEL_OPTIMIZATION=false GPU_IDX=0 FACE_DETECTION_PLUGIN=insightface.FaceDetector@retinaface_r50_v1 CALCULATION_PLUGIN=insightface.Calculator@arcface-r100-msfdrop75 EXTRA_PLUGINS=insightface.LandmarksDetector,insightface.GenderDetector,insightface.AgeDetector,insightface.facemask.MaskDetector,insightface.PoseEstimator /bin/bash -c python -m src.services.facescan.plugins.setup # buildkit

COPY tools tools # buildkit

COPY sample_images sample_images # buildkit

ARG SKIP_TESTS

COPY pytest.ini . # buildkit

RUN |8 BE_VERSION= APP_VERSION_STRING= INTEL_OPTIMIZATION=false GPU_IDX=0 FACE_DETECTION_PLUGIN=insightface.FaceDetector@retinaface_r50_v1 CALCULATION_PLUGIN=insightface.Calculator@arcface-r100-msfdrop75 EXTRA_PLUGINS=insightface.LandmarksDetector,insightface.GenderDetector,insightface.AgeDetector,insightface.facemask.MaskDetector,insightface.PoseEstimator SKIP_TESTS=1 /bin/bash -c if [ -z $SKIP_TESTS ]; then pytest -m "not performance" /app/ml/src; fi # buildkit

EXPOSE map[3000/tcp:{}]

COPY uwsgi.ini . # buildkit

CMD ["uwsgi" "--ini" "uwsgi.ini"]

Key	Value
com.nvidia.cudnn.version	8.9.0.131
maintainer	NVIDIA CORPORATION <cudatools@nvidia.com>
org.opencontainers.image.ref.name	ubuntu
org.opencontainers.image.version	20.04

Details

Container

2025-01-26 14:49:37 +01:00

OCI / Docker

linux/amd64

3.5 GiB

Versions (1) View all

1.2.0-arcface-r100-gpu

2025-01-26

compreface-core (1.2.0-arcface-r100-gpu)

Installation

Image layers

Labels