1
- ARG CUDAVER=12.2 .2
1
+ ARG CUDAVER=12.6 .2
2
2
ARG UBUNTUVER=22.04
3
3
4
4
FROM nvidia/cuda:${CUDAVER}-devel-ubuntu${UBUNTUVER} AS build
5
5
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
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
+ 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
24
15
25
16
# build and move deviceQuery to /usr/bin
26
17
RUN mkdir -p /code && \
@@ -36,21 +27,21 @@ RUN CUDA_COMPUTE_CAPABILITY=$(deviceQuery | grep Capability | head -n 1 | awk 'E
36
27
37
28
FROM ubuntu:${UBUNTUVER} AS release
38
29
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
42
33
43
34
# install va-driver
44
35
RUN apt-get update \
45
36
&& apt-get -y install libva-drm2 \
46
37
&& apt-get clean; rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/* /usr/share/doc/*
47
38
48
39
# 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
54
45
55
46
# Copy ffmpeg
56
47
COPY --from=build /app/workspace/bin/ffmpeg /usr/bin/ffmpeg
0 commit comments