1- ARG CUDAVER=12.2 .2
1+ ARG CUDAVER=12.6 .2
22ARG UBUNTUVER=22.04
33
44FROM nvidia/cuda:${CUDAVER}-devel-ubuntu${UBUNTUVER} AS build
55
6- ENV DEBIAN_FRONTEND noninteractive
7- ENV NVIDIA_VISIBLE_DEVICES all
8- ENV NVIDIA_DRIVER_CAPABILITIES compute,utility,video
6+ ENV DEBIAN_FRONTEND= noninteractive
7+ ENV NVIDIA_VISIBLE_DEVICES= all
8+ ENV NVIDIA_DRIVER_CAPABILITIES= compute,utility,video
99
10- RUN apt-get update && \
11- apt-get upgrade -y && \
12- apt-get -y --no-install-recommends install \
13- build-essential \
14- curl \
15- libva-dev \
16- python3 \
17- python-is-python3 \
18- ninja-build \
19- meson \
20- cmake \
21- git && \
22- # clean
23- apt-get clean; rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/* /usr/share/doc/*
10+ RUN apt-get update \
11+ && apt-get -y --no-install-recommends install build-essential curl ca-certificates libva-dev \
12+ python3 python-is-python3 ninja-build meson git curl \
13+ && apt-get clean; rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/* /usr/share/doc/* \
14+ && update-ca-certificates
2415
2516# build and move deviceQuery to /usr/bin
2617RUN mkdir -p /code && \
@@ -36,21 +27,21 @@ RUN CUDA_COMPUTE_CAPABILITY=$(deviceQuery | grep Capability | head -n 1 | awk 'E
3627
3728FROM ubuntu:${UBUNTUVER} AS release
3829
39- ENV DEBIAN_FRONTEND noninteractive
40- ENV NVIDIA_VISIBLE_DEVICES all
41- ENV NVIDIA_DRIVER_CAPABILITIES compute,utility,video
30+ ENV DEBIAN_FRONTEND= noninteractive
31+ ENV NVIDIA_VISIBLE_DEVICES= all
32+ ENV NVIDIA_DRIVER_CAPABILITIES= compute,utility,video
4233
4334# install va-driver
4435RUN apt-get update \
4536 && apt-get -y install libva-drm2 \
4637 && apt-get clean; rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/* /usr/share/doc/*
4738
4839# Copy libnpp
49- COPY --from=build /usr/local/cuda-12.2 /targets/x86_64-linux/lib/libnppc.so /lib/x86_64-linux-gnu/libnppc.so.12
50- COPY --from=build /usr/local/cuda-12.2 /targets/x86_64-linux/lib/libnppig.so /lib/x86_64-linux-gnu/libnppig.so.12
51- COPY --from=build /usr/local/cuda-12.2 /targets/x86_64-linux/lib/libnppicc.so /lib/x86_64-linux-gnu/libnppicc.so.12
52- COPY --from=build /usr/local/cuda-12.2 /targets/x86_64-linux/lib/libnppidei.so /lib/x86_64-linux-gnu/libnppidei.so.12
53- COPY --from=build /usr/local/cuda-12.2 /targets/x86_64-linux/lib/libnppif.so /lib/x86_64-linux-gnu/libnppif.so.12
40+ COPY --from=build /usr/local/cuda-12.6 /targets/x86_64-linux/lib/libnppc.so /lib/x86_64-linux-gnu/libnppc.so.12
41+ COPY --from=build /usr/local/cuda-12.6 /targets/x86_64-linux/lib/libnppig.so /lib/x86_64-linux-gnu/libnppig.so.12
42+ COPY --from=build /usr/local/cuda-12.6 /targets/x86_64-linux/lib/libnppicc.so /lib/x86_64-linux-gnu/libnppicc.so.12
43+ COPY --from=build /usr/local/cuda-12.6 /targets/x86_64-linux/lib/libnppidei.so /lib/x86_64-linux-gnu/libnppidei.so.12
44+ COPY --from=build /usr/local/cuda-12.6 /targets/x86_64-linux/lib/libnppif.so /lib/x86_64-linux-gnu/libnppif.so.12
5445
5546# Copy ffmpeg
5647COPY --from=build /app/workspace/bin/ffmpeg /usr/bin/ffmpeg
0 commit comments