File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -16,7 +16,7 @@ ENV GOSU_VERSION=1.10
1616
1717# Install all basic requirements
1818RUN \
19- curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/rhel7 /x86_64/D42D0685.pub | sed '/^Version/d' \
19+ curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/rhel8 /x86_64/D42D0685.pub | sed '/^Version/d' \
2020 > /etc/pki/rpm-gpg/RPM-GPG-KEY-NVIDIA && \
2121 dnf -y update && \
2222 dnf -y install dnf-plugins-core && \
Original file line number Diff line number Diff line change @@ -18,7 +18,7 @@ ENV GOSU_VERSION=1.10
1818
1919# Install all basic requirements
2020RUN \
21- curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/rhel7 /x86_64/D42D0685.pub | sed '/^Version/d' \
21+ curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/rhel8 /x86_64/D42D0685.pub | sed '/^Version/d' \
2222 > /etc/pki/rpm-gpg/RPM-GPG-KEY-NVIDIA && \
2323 dnf -y update && \
2424 dnf -y install dnf-plugins-core && \
Original file line number Diff line number Diff line change @@ -17,7 +17,7 @@ ENV GOSU_VERSION=1.10
1717
1818# Install all basic requirements
1919RUN \
20- curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/rhel7 /x86_64/D42D0685.pub | sed '/^Version/d' \
20+ curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/rhel8 /x86_64/D42D0685.pub | sed '/^Version/d' \
2121 > /etc/pki/rpm-gpg/RPM-GPG-KEY-NVIDIA && \
2222 dnf -y update && \
2323 dnf -y install dnf-plugins-core && \
3434# NCCL2 (License: https://docs.nvidia.com/deeplearning/sdk/nccl-sla/index.html)
3535RUN \
3636 export NCCL_VERSION=$NCCL_VERSION && \
37- dnf config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel7 /x86_64/cuda-rhel7 .repo && \
37+ dnf config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel8 /x86_64/cuda-rhel8 .repo && \
3838 dnf -y update && \
3939 dnf install -y libnccl-${NCCL_VERSION}+cuda12.9 libnccl-devel-${NCCL_VERSION}+cuda12.9
4040
Original file line number Diff line number Diff line change 3636# NCCL2 (License: https://docs.nvidia.com/deeplearning/sdk/nccl-sla/index.html)
3737RUN \
3838 export NCCL_VERSION=$NCCL_VERSION && \
39- dnf config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel7 /x86_64/cuda-rhel7 .repo && \
39+ dnf config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel8 /x86_64/cuda-rhel8 .repo && \
4040 dnf -y update && \
4141 dnf install -y libnccl-${NCCL_VERSION}+cuda12.9 libnccl-devel-${NCCL_VERSION}+cuda12.9 libnccl-static-${NCCL_VERSION}+cuda12.9
4242
You can’t perform that action at this time.
0 commit comments