From c72de59bd02f69d4becd64d73c9df556c4b6c0f4 Mon Sep 17 00:00:00 2001
From: Christopher Desiniotis <cdesiniotis@nvidia.com>
Date: Mon, 8 Jul 2024 11:02:42 -0700
Subject: [PATCH] Switch to using nvcr.io for CUDA base images

Signed-off-by: Christopher Desiniotis <cdesiniotis@nvidia.com>
---
 base/Dockerfile                     | 4 ++--
 centos7/Dockerfile                  | 2 +-
 centos8/Dockerfile                  | 2 +-
 coreos/Dockerfile                   | 2 +-
 fedora/Dockerfile                   | 2 +-
 flatcar/Dockerfile                  | 2 +-
 photon3.0/Dockerfile                | 2 +-
 rhel7/Dockerfile                    | 2 +-
 rhel8/Dockerfile                    | 4 ++--
 rhel8/precompiled/Dockerfile        | 2 +-
 rhel9/Dockerfile                    | 4 ++--
 sle15/Dockerfile                    | 2 +-
 ubuntu16.04/Dockerfile              | 2 +-
 ubuntu18.04/Dockerfile              | 2 +-
 ubuntu20.04/Dockerfile              | 4 ++--
 ubuntu20.04/precompiled/Dockerfile  | 2 +-
 ubuntu22.04/Dockerfile              | 4 ++--
 ubuntu22.04/precompiled/Dockerfile  | 2 +-
 vgpu-manager/rhel8/Dockerfile       | 2 +-
 vgpu-manager/ubuntu20.04/Dockerfile | 2 +-
 vgpu-manager/ubuntu22.04/Dockerfile | 2 +-
 21 files changed, 26 insertions(+), 26 deletions(-)

diff --git a/base/Dockerfile b/base/Dockerfile
index 82d4519f..27118eba 100644
--- a/base/Dockerfile
+++ b/base/Dockerfile
@@ -1,7 +1,7 @@
 ARG CUDA_VERSION
 
 # Ubuntu 22.04
-FROM nvidia/cuda:${CUDA_VERSION}-base-ubuntu22.04 as jammy
+FROM nvcr.io/nvidia/cuda:${CUDA_VERSION}-base-ubuntu22.04 as jammy
 
 SHELL ["/bin/bash", "-c"]
 
@@ -36,7 +36,7 @@ RUN chmod +x /usr/local/bin/generate-ci-config && \
 ENTRYPOINT ["/usr/bin/sleep","1000"]
 
 # Ubuntu 20.04
-FROM nvidia/cuda:${CUDA_VERSION}-base-ubuntu20.04 as focal
+FROM nvcr.io/nvidia/cuda:${CUDA_VERSION}-base-ubuntu20.04 as focal
 
 SHELL ["/bin/bash", "-c"]
 
diff --git a/centos7/Dockerfile b/centos7/Dockerfile
index 1f8a6afc..06641257 100644
--- a/centos7/Dockerfile
+++ b/centos7/Dockerfile
@@ -1,5 +1,5 @@
 ARG CUDA_VERSION
-FROM nvidia/cuda:${CUDA_VERSION}-base-centos7
+FROM nvcr.io/nvidia/cuda:${CUDA_VERSION}-base-centos7
 
 ENV NVIDIA_VISIBLE_DEVICES=void
 
diff --git a/centos8/Dockerfile b/centos8/Dockerfile
index 2dbdb66e..7b08016f 100644
--- a/centos8/Dockerfile
+++ b/centos8/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.4.1-base-centos8
+FROM nvcr.io/nvidia/cuda:11.4.1-base-centos8
 
 ENV NVIDIA_VISIBLE_DEVICES=void
 
diff --git a/coreos/Dockerfile b/coreos/Dockerfile
index 271510a0..9bae661f 100644
--- a/coreos/Dockerfile
+++ b/coreos/Dockerfile
@@ -1,5 +1,5 @@
 ARG CUDA_VERSION=11.7.0
-FROM nvidia/cuda:$CUDA_VERSION-base-ubi8 as license
+FROM nvcr.io/nvidia/cuda:$CUDA_VERSION-base-ubi8 as license
 
 # Build nvidia-container-runtime binary
 FROM golang:1.11.2 as build-runtime
diff --git a/fedora/Dockerfile b/fedora/Dockerfile
index 44841d34..e1e96d25 100644
--- a/fedora/Dockerfile
+++ b/fedora/Dockerfile
@@ -1,7 +1,7 @@
 ARG FEDORA_VERSION=36
 ARG CUDA_VERSION
 
-FROM nvidia/cuda:${CUDA_VERSION}-base-ubi8 as build
+FROM nvcr.io/nvidia/cuda:${CUDA_VERSION}-base-ubi8 as build
 
 ARG TARGETARCH
 
diff --git a/flatcar/Dockerfile b/flatcar/Dockerfile
index 810fd09d..1d14acea 100644
--- a/flatcar/Dockerfile
+++ b/flatcar/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.7.1-base-ubuntu22.04
+FROM nvcr.io/nvidia/cuda:11.7.1-base-ubuntu22.04
 
 ENV NVIDIA_VISIBLE_DEVICES=void
 
diff --git a/photon3.0/Dockerfile b/photon3.0/Dockerfile
index 0103f924..f4752a76 100644
--- a/photon3.0/Dockerfile
+++ b/photon3.0/Dockerfile
@@ -1,5 +1,5 @@
 ARG CUDA_VERSION=11.7.0
-FROM nvidia/cuda:$CUDA_VERSION-base-ubi8 as license
+FROM nvcr.io/nvidia/cuda:$CUDA_VERSION-base-ubi8 as license
 
 FROM photon:3.0
 
diff --git a/rhel7/Dockerfile b/rhel7/Dockerfile
index ffa3e32d..d72c3308 100644
--- a/rhel7/Dockerfile
+++ b/rhel7/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.4.1-base-ubi7
+FROM nvcr.io/nvidia/cuda:11.4.1-base-ubi7
 
 ENV NVIDIA_VISIBLE_DEVICES=void
 
diff --git a/rhel8/Dockerfile b/rhel8/Dockerfile
index 847abda0..bd5f0de5 100644
--- a/rhel8/Dockerfile
+++ b/rhel8/Dockerfile
@@ -1,5 +1,5 @@
 ARG CUDA_VERSION
-FROM nvidia/cuda:${CUDA_VERSION}-base-ubi8 as build
+FROM nvcr.io/nvidia/cuda:${CUDA_VERSION}-base-ubi8 as build
 
 ARG TARGETARCH
 
@@ -23,7 +23,7 @@ RUN git clone https://github.com/NVIDIA/gpu-driver-container driver && \
     go build -o vgpu-util && \
     mv vgpu-util /work
 
-FROM nvidia/cuda:${CUDA_VERSION}-base-ubi8
+FROM nvcr.io/nvidia/cuda:${CUDA_VERSION}-base-ubi8
 
 ARG TARGETARCH
 ENV TARGETARCH=$TARGETARCH
diff --git a/rhel8/precompiled/Dockerfile b/rhel8/precompiled/Dockerfile
index 7f34858c..5563a500 100644
--- a/rhel8/precompiled/Dockerfile
+++ b/rhel8/precompiled/Dockerfile
@@ -50,7 +50,7 @@ RUN export KVER=$(echo ${KERNEL_VERSION} | cut -d '-' -f 1) \
         --define "driver_branch ${DRIVER_STREAM}" \
         -v -bb SPECS/kmod-nvidia.spec
 
-FROM nvidia/cuda:${CUDA_VERSION}-base-${CUDA_DIST}
+FROM nvcr.io/nvidia/cuda:${CUDA_VERSION}-base-${CUDA_DIST}
 
 ARG KERNEL_VERSION=''
 ARG RHEL_VERSION=''
diff --git a/rhel9/Dockerfile b/rhel9/Dockerfile
index be735551..e932fd4e 100644
--- a/rhel9/Dockerfile
+++ b/rhel9/Dockerfile
@@ -1,5 +1,5 @@
 ARG CUDA_VERSION
-FROM nvidia/cuda:${CUDA_VERSION}-base-ubi9 as build
+FROM nvcr.io/nvidia/cuda:${CUDA_VERSION}-base-ubi9 as build
 
 ARG TARGETARCH
 
@@ -23,7 +23,7 @@ RUN git clone https://github.com/NVIDIA/gpu-driver-container driver && \
     go build -o vgpu-util && \
     mv vgpu-util /work
 
-FROM nvidia/cuda:${CUDA_VERSION}-base-ubi9
+FROM nvcr.io/nvidia/cuda:${CUDA_VERSION}-base-ubi9
 
 ARG TARGETARCH
 ENV TARGETARCH=$TARGETARCH
diff --git a/sle15/Dockerfile b/sle15/Dockerfile
index f80b144f..749f0295 100644
--- a/sle15/Dockerfile
+++ b/sle15/Dockerfile
@@ -1,6 +1,6 @@
 ARG SLES_VERSION
 ARG CUDA_VERSION=11.7.0
-FROM nvidia/cuda:$CUDA_VERSION-base-ubi8 as license
+FROM nvcr.io/nvidia/cuda:$CUDA_VERSION-base-ubi8 as license
 
 FROM registry.suse.com/bci/golang:1.17 as build
 
diff --git a/ubuntu16.04/Dockerfile b/ubuntu16.04/Dockerfile
index ccb23ec1..562d7e0b 100644
--- a/ubuntu16.04/Dockerfile
+++ b/ubuntu16.04/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.3.0-base-ubuntu16.04
+FROM nvcr.io/nvidia/cuda:11.3.0-base-ubuntu16.04
 
 ENV NVIDIA_VISIBLE_DEVICES=void
 
diff --git a/ubuntu18.04/Dockerfile b/ubuntu18.04/Dockerfile
index 41cbde75..1333663d 100644
--- a/ubuntu18.04/Dockerfile
+++ b/ubuntu18.04/Dockerfile
@@ -1,5 +1,5 @@
 ARG CUDA_VERSION
-FROM nvidia/cuda:${CUDA_VERSION}-base-ubuntu18.04
+FROM nvcr.io/nvidia/cuda:${CUDA_VERSION}-base-ubuntu18.04
 
 #ARG BASE_URL=http://us.download.nvidia.com/XFree86/Linux-x86_64
 ARG BASE_URL=https://us.download.nvidia.com/tesla
diff --git a/ubuntu20.04/Dockerfile b/ubuntu20.04/Dockerfile
index d9b61c08..339c8055 100644
--- a/ubuntu20.04/Dockerfile
+++ b/ubuntu20.04/Dockerfile
@@ -1,5 +1,5 @@
 ARG CUDA_VERSION
-FROM nvidia/cuda:${CUDA_VERSION}-base-ubuntu20.04 as build
+FROM nvcr.io/nvidia/cuda:${CUDA_VERSION}-base-ubuntu20.04 as build
 
 ARG TARGETARCH
 
@@ -34,7 +34,7 @@ RUN git clone https://github.com/NVIDIA/gpu-driver-container driver && \
     go build -o vgpu-util && \
     mv vgpu-util /work
 
-FROM nvidia/cuda:${CUDA_VERSION}-base-ubuntu20.04
+FROM nvcr.io/nvidia/cuda:${CUDA_VERSION}-base-ubuntu20.04
 
 SHELL ["/bin/bash", "-c"]
 
diff --git a/ubuntu20.04/precompiled/Dockerfile b/ubuntu20.04/precompiled/Dockerfile
index 66e63402..7d1554bd 100644
--- a/ubuntu20.04/precompiled/Dockerfile
+++ b/ubuntu20.04/precompiled/Dockerfile
@@ -1,5 +1,5 @@
 ARG CUDA_VERSION
-FROM nvidia/cuda:${CUDA_VERSION}-base-ubuntu20.04
+FROM nvcr.io/nvidia/cuda:${CUDA_VERSION}-base-ubuntu20.04
 
 ENV DEBIAN_FRONTEND=noninteractive
 
diff --git a/ubuntu22.04/Dockerfile b/ubuntu22.04/Dockerfile
index 57d09664..2670a4f5 100644
--- a/ubuntu22.04/Dockerfile
+++ b/ubuntu22.04/Dockerfile
@@ -1,5 +1,5 @@
 ARG CUDA_VERSION
-FROM nvidia/cuda:${CUDA_VERSION}-base-ubuntu22.04 as build
+FROM nvcr.io/nvidia/cuda:${CUDA_VERSION}-base-ubuntu22.04 as build
 
 ARG TARGETARCH
 
@@ -34,7 +34,7 @@ RUN git clone https://github.com/NVIDIA/gpu-driver-container driver && \
     go build -o vgpu-util && \
     mv vgpu-util /work
 
-FROM nvidia/cuda:${CUDA_VERSION}-base-ubuntu22.04
+FROM nvcr.io/nvidia/cuda:${CUDA_VERSION}-base-ubuntu22.04
 
 SHELL ["/bin/bash", "-c"]
 
diff --git a/ubuntu22.04/precompiled/Dockerfile b/ubuntu22.04/precompiled/Dockerfile
index 4dfcdbc0..a0d91f68 100644
--- a/ubuntu22.04/precompiled/Dockerfile
+++ b/ubuntu22.04/precompiled/Dockerfile
@@ -1,5 +1,5 @@
 ARG CUDA_VERSION
-FROM nvidia/cuda:${CUDA_VERSION}-base-ubuntu22.04
+FROM nvcr.io/nvidia/cuda:${CUDA_VERSION}-base-ubuntu22.04
 
 ENV DEBIAN_FRONTEND=noninteractive
 
diff --git a/vgpu-manager/rhel8/Dockerfile b/vgpu-manager/rhel8/Dockerfile
index 49cac7be..75754acd 100644
--- a/vgpu-manager/rhel8/Dockerfile
+++ b/vgpu-manager/rhel8/Dockerfile
@@ -1,5 +1,5 @@
 ARG CUDA_VERSION
-FROM nvidia/cuda:${CUDA_VERSION}-base-ubi8
+FROM nvcr.io/nvidia/cuda:${CUDA_VERSION}-base-ubi8
 
 ARG DRIVER_VERSION
 ENV DRIVER_VERSION=$DRIVER_VERSION
diff --git a/vgpu-manager/ubuntu20.04/Dockerfile b/vgpu-manager/ubuntu20.04/Dockerfile
index e74c8ab3..c946a5be 100644
--- a/vgpu-manager/ubuntu20.04/Dockerfile
+++ b/vgpu-manager/ubuntu20.04/Dockerfile
@@ -1,5 +1,5 @@
 ARG CUDA_VERSION
-FROM nvidia/cuda:${CUDA_VERSION}-base-ubuntu20.04
+FROM nvcr.io/nvidia/cuda:${CUDA_VERSION}-base-ubuntu20.04
 
 ARG DRIVER_VERSION
 ENV DRIVER_VERSION=$DRIVER_VERSION
diff --git a/vgpu-manager/ubuntu22.04/Dockerfile b/vgpu-manager/ubuntu22.04/Dockerfile
index 5bc98872..2bee7f0d 100644
--- a/vgpu-manager/ubuntu22.04/Dockerfile
+++ b/vgpu-manager/ubuntu22.04/Dockerfile
@@ -1,5 +1,5 @@
 ARG CUDA_VERSION
-FROM nvidia/cuda:${CUDA_VERSION}-base-ubuntu22.04
+FROM nvcr.io/nvidia/cuda:${CUDA_VERSION}-base-ubuntu22.04
 
 ARG DRIVER_VERSION
 ENV DRIVER_VERSION=$DRIVER_VERSION