| 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:{}] |
| WORKDIR /app |
| EXPOSE map[3333/tcp:{}] |
| ADD . /app # buildkit |
| RUN /bin/sh -c pip install --upgrade pip # buildkit |
| RUN /bin/sh -c pip install -r requirements.txt # buildkit |
| RUN /bin/sh -c echo 'Downloading the latest failure detection AI model in Darknet format...' # buildkit |
| RUN /bin/sh -c wget -O model/model-weights.darknet $(cat model/model-weights.darknet.url | tr -d '\r') # buildkit |
| RUN /bin/sh -c echo 'Downloading the latest failure detection AI model in ONNX format...' # buildkit |
| RUN /bin/sh -c wget -O model/model-weights.onnx $(cat model/model-weights.onnx.url | tr -d '\r') # buildkit |
