rswafford
Repositories
2
 Projects Packages
8
 Public activity Starred repositories

obico-ml-api (sha-6f93896)

Published 2026-11-04 08:26:27 +00:00 by rswafford in rswafford/obico-images

Installation

docker pull git.atomic-homelab.com/rswafford/obico-ml-api:sha-6f93896
sha256:9f8d7c5702fd19d9490c1f75500ed3aeea71bd33202b62b5d4d3a6af7b6c2baf

Image layers

ARG RELEASE
ARG LAUNCHPAD_BUILD_ARCH
LABEL org.opencontainers.image.ref.name=ubuntu
LABEL org.opencontainers.image.version=20.04
ADD file:4809da414c2d478b4d991cbdaa2df457f2b3d07d0ff6cf673f09a66f90833e81 in /
CMD ["/bin/bash"]
ENV NVARCH=x86_64
ENV NVIDIA_REQUIRE_CUDA=cuda>=11.4
ENV NV_CUDA_CUDART_VERSION=11.4.148-1
ENV NV_CUDA_COMPAT_PACKAGE=cuda-compat-11-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=11.4.3
RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends cuda-cudart-11-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=11.4.3-1
ENV NV_NVTX_VERSION=11.4.120-1
ENV NV_LIBNPP_VERSION=11.4.0.110-1
ENV NV_LIBNPP_PACKAGE=libnpp-11-4=11.4.0.110-1
ENV NV_LIBCUSPARSE_VERSION=11.6.0.120-1
ENV NV_LIBCUBLAS_PACKAGE_NAME=libcublas-11-4
ENV NV_LIBCUBLAS_VERSION=11.6.5.2-1
ENV NV_LIBCUBLAS_PACKAGE=libcublas-11-4=11.6.5.2-1
ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2
ENV NV_LIBNCCL_PACKAGE_VERSION=2.11.4-1
ENV NCCL_VERSION=2.11.4-1
ENV NV_LIBNCCL_PACKAGE=libnccl2=2.11.4-1+cuda11.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-11-4=${NV_CUDA_LIB_VERSION} ${NV_LIBNPP_PACKAGE} cuda-nvtx-11-4=${NV_NVTX_VERSION} libcusparse-11-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_CUDNN_VERSION=8.2.4.15
ENV NV_CUDNN_PACKAGE_NAME=libcudnn8
ENV NV_CUDNN_PACKAGE=libcudnn8=8.2.4.15-1+cuda11.4
ARG TARGETARCH
LABEL maintainer=NVIDIA CORPORATION <cudatools@nvidia.com>
LABEL com.nvidia.cudnn.version=8.2.4.15
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
COPY /darknet /darknet # buildkit
WORKDIR /app
RUN /bin/sh -c apt update && apt install --no-install-recommends -y ca-certificates python3-pip wget # buildkit
RUN /bin/sh -c pip install --upgrade pip # buildkit
RUN /bin/sh -c pip install opencv_python_headless onnxruntime-gpu # buildkit
ADD requirements.txt . # buildkit
RUN /bin/sh -c pip install -r requirements.txt # buildkit
EXPOSE map[3333/tcp:{}]
RUN /bin/sh -c apt update && apt install -y curl # buildkit
WORKDIR /app
EXPOSE [3333/tcp]
COPY requirements.txt /app/ # buildkit
COPY model/*.url model/*.sha256 /app/model/ # buildkit
COPY scripts/fetch_verified.sh /usr/local/bin/fetch_verified # buildkit
RUN /bin/sh -c pip install --no-cache-dir --upgrade pip && pip install --no-cache-dir -r requirements.txt # buildkit
RUN /bin/sh -c mkdir -p /model_cache/ml_api/darknet /model_cache/ml_api/onnx /model_cache/ml_api/rknn && if [ -e /darknet/libdarknet_gpu.so ] || [ -e /darknet/libdarknet_cpu.so ]; then echo 'Downloading the latest failure detection AI model in Darknet format...'; fetch_verified "$(tr -d '\r' < model/model-weights.darknet.url)" /model_cache/ml_api/darknet/model-weights.darknet 50000000 "$(tr -d '\r' < model/model-weights.darknet.sha256)"; else echo 'No darknet library in this base; skipping its 200 MB of weights, which lib/darknet.py could not load here.'; fi && if python3 -c 'import onnxruntime' 2>/dev/null; then echo 'Downloading the latest failure detection AI model in ONNX format...'; fetch_verified "$(tr -d '\r' < model/model-weights.onnx.url)" /model_cache/ml_api/onnx/model-weights.onnx 50000000 "$(tr -d '\r' < model/model-weights.onnx.sha256)"; else echo 'No onnxruntime in this base; skipping its 200 MB of weights, which lib/onnx.py could not load here.'; fi && if python3 -c 'import rknnlite.api' 2>/dev/null; then echo 'Downloading the latest failure detection AI model in RKNN format...'; fetch_verified "$(tr -d '\r' < model/model-weights.rknn.url)" /model_cache/ml_api/rknn/model-weights.rknn 50000000 "$(tr -d '\r' < model/model-weights.rknn.sha256)"; else echo 'No rknn-toolkit-lite2 in this base; skipping the RKNN model, which lib/rknn.py could not load here.'; fi # buildkit
COPY . /app # buildkit

Labels

Key Value
com.nvidia.cudnn.version 8.2.4.15
maintainer NVIDIA CORPORATION <cudatools@nvidia.com>
org.opencontainers.image.ref.name ubuntu
org.opencontainers.image.revision 6f93896
org.opencontainers.image.source http://forgejo.default.svc.cluster.local.:3000/rswafford/obico-images
org.opencontainers.image.version 20.04
Details
Container
rswafford/obico-images
2026-11-04 08:26:27 +00:00
10
OCI / Docker
linux/amd64
Repository website
3 GiB
Versions (6) View all
2026.10.02 2026-11-04
sha-6f93896 2026-11-04
release 2026-11-04
latest 2026-11-04
sha-49c0bc7 2026-10-02
Issues