@@ -7,20 +7,7 @@ ENV DEBIAN_FRONTEND=noninteractive
77ENV NVIDIA_VISIBLE_DEVICES=all
88ENV 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+
2411
2512# build and move deviceQuery to /usr/bin
2613RUN mkdir -p /code && \
@@ -46,11 +33,11 @@ RUN apt-get update \
4633 && apt-get clean; rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/* /usr/share/doc/*
4734
4835# 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
36+ COPY --from=build /usr/local/cuda-12.6 /targets/x86_64-linux/lib/libnppc.so /lib/x86_64-linux-gnu/libnppc.so.12
37+ COPY --from=build /usr/local/cuda-12.6 /targets/x86_64-linux/lib/libnppig.so /lib/x86_64-linux-gnu/libnppig.so.12
38+ COPY --from=build /usr/local/cuda-12.6 /targets/x86_64-linux/lib/libnppicc.so /lib/x86_64-linux-gnu/libnppicc.so.12
39+ COPY --from=build /usr/local/cuda-12.6 /targets/x86_64-linux/lib/libnppidei.so /lib/x86_64-linux-gnu/libnppidei.so.12
40+ COPY --from=build /usr/local/cuda-12.6 /targets/x86_64-linux/lib/libnppif.so /lib/x86_64-linux-gnu/libnppif.so.12
5441
5542# Copy ffmpeg
5643COPY --from=build /app/workspace/bin/ffmpeg /usr/bin/ffmpeg
0 commit comments