| ARG RELEASE |
| ARG LAUNCHPAD_BUILD_ARCH |
| LABEL org.opencontainers.image.ref.name=ubuntu |
| LABEL org.opencontainers.image.version=22.04 |
| ADD file:415bbc01dfb447d002e2d8173e113ef025d2bbfa20f1205823fa699dc87a2019 in / |
| CMD ["/bin/bash"] |
| ARG PYTORCH_VERSION=2.8.0 |
| ARG TRITON_VERSION= |
| ARG TARGETPLATFORM=linux/amd64 |
| ARG CUDA_VERSION=12.6.3 |
| LABEL com.nvidia.volumes.needed=nvidia_driver |
| RUN |4 PYTORCH_VERSION=2.8.0 TRITON_VERSION= TARGETPLATFORM=linux/amd64 CUDA_VERSION=12.6.3 /bin/sh -c apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends ca-certificates libjpeg-dev libpng-dev && rm -rf /var/lib/apt/lists/* # buildkit |
| COPY /opt/conda /opt/conda # buildkit |
| RUN |4 PYTORCH_VERSION=2.8.0 TRITON_VERSION= TARGETPLATFORM=linux/amd64 CUDA_VERSION=12.6.3 /bin/sh -c if test -n "${TRITON_VERSION}" -a "${TARGETPLATFORM}" != "linux/arm64"; then DEBIAN_FRONTEND=noninteractive apt install -y --no-install-recommends gcc; rm -rf /var/lib/apt/lists/*; fi # buildkit |
| ENV PATH=/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| ENV NVIDIA_VISIBLE_DEVICES=all |
| ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility |
| ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64 |
| ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| ENV PYTORCH_VERSION=2.8.0 |
| WORKDIR /workspace |
| RUN /bin/sh -c apt-get update && apt-get install -y build-essential cmake git pkg-config wget unzip yasm nasm libtool libc6 libc6-dev libnuma1 libnuma-dev gcc g++ libc6-dev linux-libc-dev libgl1-mesa-glx libglib2.0-0 libgomp1 libjpeg-dev libpng-dev libx264-dev libx265-dev libvpx-dev libmp3lame-dev libv4l-dev libturbojpeg0-dev python3-dev python3-numpy && rm -rf /var/lib/apt/lists/* # buildkit |
| RUN /bin/sh -c apt-get update && apt-get install -y wget gnupg && wget -O - https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/3bf863cc.pub | apt-key add - && echo "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64 /" > /etc/apt/sources.list.d/cuda.list && apt-get update && apt-get install -y cuda-nvcc-12-6 cuda-cudart-dev-12-6 libnpp-dev-12-6 && apt-get remove -y wget gnupg && apt-get autoremove -y && rm -rf /var/lib/apt/lists/* # buildkit |
| ENV PATH=/usr/local/cuda/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/nvidia/lib:/usr/local/nvidia/lib64 |
| RUN /bin/sh -c cd /tmp && git clone https://git.videolan.org/git/ffmpeg/nv-codec-headers.git && cd nv-codec-headers && make install && cd / && rm -rf /tmp/* # buildkit |
| RUN /bin/sh -c cd /tmp && echo "Building FFmpeg with NVIDIA CUDA support..." && git clone https://git.ffmpeg.org/ffmpeg.git ffmpeg/ && cd ffmpeg && ./configure --prefix=/usr/local --enable-shared --disable-static --enable-nonfree --enable-gpl --enable-cuda-nvcc --enable-cuvid --enable-nvdec --enable-nvenc --enable-libnpp --extra-cflags=-I/usr/local/cuda/include --extra-ldflags=-L/usr/local/cuda/lib64 --enable-libx264 --enable-libx265 --enable-libvpx --enable-libmp3lame && make -j$(nproc) && make install && ldconfig && echo "=== Verifying FFmpeg CUVID Support ===" && (ffmpeg -hide_banner -decoders 2>/dev/null | grep cuvid || echo "No CUVID decoders found") && echo "=== Verifying FFmpeg NVENC Support ===" && (ffmpeg -hide_banner -encoders 2>/dev/null | grep nvenc || echo "No NVENC encoders found") && cd / && rm -rf /tmp/* # buildkit |
| ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/lib:/usr/local/cuda/lib64:/usr/local/nvidia/lib:/usr/local/nvidia/lib64 |
| ENV PKG_CONFIG_PATH=/usr/local/lib/pkgconfig: |
| ENV PYTHONPATH=/usr/local/lib/python3.10/dist-packages: |
| ENV OPENCV_FFMPEG_CAPTURE_OPTIONS=rtsp_transport;tcp|hwaccel;cuda|hwaccel_device;0|video_codec;h264_cuvid|hwaccel_output_format;cuda |
| ENV OPENCV_FFMPEG_WRITER_OPTIONS=video_codec;h264_nvenc|preset;fast|tune;zerolatency|gpu;0 |
| ENV CUDA_VISIBLE_DEVICES=0 |
| ENV NVIDIA_VISIBLE_DEVICES=all |
| ENV NVIDIA_DRIVER_CAPABILITIES=compute,video,utility |
| COPY requirements.base.txt . # buildkit |
| RUN /bin/sh -c grep -v opencv-python requirements.base.txt > requirements.tmp && mv requirements.tmp requirements.base.txt && pip install --no-cache-dir -r requirements.base.txt # buildkit |
| WORKDIR /app |
| RUN /bin/sh -c mkdir -p /app/images && chmod 755 /app/images # buildkit |
| CMD ["python3" "-m" "fastapi" "run" "--host" "0.0.0.0" "--port" "8000"] |
