Skip to content

Commit 680b398

Browse files
committed
Cuda Rocky fix
1 parent def2057 commit 680b398

File tree

1 file changed

+6
-29
lines changed

1 file changed

+6
-29
lines changed

cuda-rocky.dockerfile

Lines changed: 6 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -1,33 +1,10 @@
11
ARG VER=8
22

3-
FROM nvidia/cuda:11.6.0-devel-centos${VER} AS build
3+
FROM nvidia/cuda:11.7.0-devel-rockylinux${VER} AS build
44

55
ENV NVIDIA_VISIBLE_DEVICES all
66
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility,video
77

8-
9-
# Fix EOL Mirror issue
10-
RUN sed -i 's/mirrorlist/#mirrorlist/g' /etc/yum.repos.d/CentOS-Linux-*
11-
# Pick one of these mirrors that works. vault.centos.org was giving me timeout issues
12-
# RUN sed -i 's|#baseurl=http://mirror.centos.org|baseurl=http://vault.centos.org|g' /etc/yum.repos.d/CentOS-Linux-*
13-
# RUN sed -i 's|#baseurl=http://mirror.centos.org|baseurl=http://linuxsoft.cern.ch/centos-vault/|g' /etc/yum.repos.d/CentOS-Linux-*
14-
# RUN sed -i 's|#baseurl=http://mirror.centos.org|baseurl=http://archive.kernel.org/centos-vault/|g' /etc/yum.repos.d/CentOS-Linux-*
15-
RUN sed -i 's|#baseurl=http://mirror.centos.org|baseurl=http://mirror.nsc.liu.se/centos-store|g' /etc/yum.repos.d/CentOS-Linux-*
16-
17-
# upgrade to Rocky Linux 8
18-
RUN dnf install -y ncurses \
19-
&& dnf clean all \
20-
&& rm -rf /var/cache/dnf \
21-
&& rm -rf /var/cache/yum \
22-
&& dnf install -y dnf-plugins-core \
23-
&& dnf -y update
24-
25-
WORKDIR /opt/conversion_to_rocky
26-
RUN curl -O https://raw.githubusercontent.com/rocky-linux/rocky-tools/main/migrate2rocky/migrate2rocky.sh && \
27-
chmod +x migrate2rocky.sh && \
28-
bash migrate2rocky.sh -r
29-
30-
318
RUN yum group install -y "Development Tools" \
329
&& yum install -y curl libva-devel python3 \
3310
&& yum install -y meson ninja-build --enablerepo=powertools \
@@ -52,11 +29,11 @@ RUN yum install -y libva \
5229
&& yum clean all
5330

5431
# Copy libnpp
55-
COPY --from=build /usr/local/cuda-11.6/targets/x86_64-linux/lib/libnppc.so.11 /lib64/libnppc.so.11
56-
COPY --from=build /usr/local/cuda-11.6/targets/x86_64-linux/lib/libnppig.so.11 /lib64/libnppig.so.11
57-
COPY --from=build /usr/local/cuda-11.6/targets/x86_64-linux/lib/libnppicc.so.11 /lib64/libnppicc.so.11
58-
COPY --from=build /usr/local/cuda-11.6/targets/x86_64-linux/lib/libnppidei.so.11 /lib64/libnppidei.so.11
59-
COPY --from=build /usr/local/cuda-11.6/targets/x86_64-linux/lib/libnppif.so.11 /lib64/libnppif.so.11
32+
COPY --from=build /usr/local/cuda-11.7/targets/x86_64-linux/lib/libnppc.so.11 /lib64/libnppc.so.11
33+
COPY --from=build /usr/local/cuda-11.7/targets/x86_64-linux/lib/libnppig.so.11 /lib64/libnppig.so.11
34+
COPY --from=build /usr/local/cuda-11.7/targets/x86_64-linux/lib/libnppicc.so.11 /lib64/libnppicc.so.11
35+
COPY --from=build /usr/local/cuda-11.7/targets/x86_64-linux/lib/libnppidei.so.11 /lib64/libnppidei.so.11
36+
COPY --from=build /usr/local/cuda-11.7/targets/x86_64-linux/lib/libnppif.so.11 /lib64/libnppif.so.11
6037

6138
# Copy ffmpeg
6239
COPY --from=build /app/workspace/bin/ffmpeg /usr/bin/ffmpeg

0 commit comments

Comments
 (0)