diff --git a/ci-wheel.Dockerfile b/ci-wheel.Dockerfile index 848bae5..d21e448 100644 --- a/ci-wheel.Dockerfile +++ b/ci-wheel.Dockerfile @@ -2,7 +2,7 @@ ARG CUDA_VER=11.8.0 ARG LINUX_VER=ubuntu20.04 ARG REAL_ARCH=x86_64 -ARG BASE_IMAGE=nvidia/cuda:${CUDA_VER}-devel-${LINUX_VER} +ARG BASE_IMAGE=nvcr.io/nvidia/cuda:${CUDA_VER}-devel-${LINUX_VER} FROM ${BASE_IMAGE} ARG CUDA_VER diff --git a/citestwheel.Dockerfile b/citestwheel.Dockerfile index 061b5a4..cb0e586 100644 --- a/citestwheel.Dockerfile +++ b/citestwheel.Dockerfile @@ -1,7 +1,7 @@ ARG CUDA_VER=11.8.0 ARG LINUX_VER=ubuntu18.04 -ARG BASE_IMAGE=nvidia/cuda:${CUDA_VER}-devel-${LINUX_VER} +ARG BASE_IMAGE=nvcr.io/nvidia/cuda:${CUDA_VER}-devel-${LINUX_VER} FROM ${BASE_IMAGE} ARG CUDA_VER