@@ -7,20 +7,7 @@ ENV DEBIAN_FRONTEND=noninteractive
7
7
ENV NVIDIA_VISIBLE_DEVICES=all
8
8
ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility,video
9
9
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
+
24
11
25
12
# build and move deviceQuery to /usr/bin
26
13
RUN mkdir -p /code && \
@@ -46,11 +33,11 @@ RUN apt-get update \
46
33
&& apt-get clean; rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/* /usr/share/doc/*
47
34
48
35
# 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
54
41
55
42
# Copy ffmpeg
56
43
COPY --from=build /app/workspace/bin/ffmpeg /usr/bin/ffmpeg
0 commit comments