Pull nvidia cuda images from public ecr

satishpasumarthi · satishpasumarthi · commit 997d0209675c · 2022-02-07T14:05:48.000-08:00
diff --git a/docker/1.15/Dockerfile.gpu b/docker/1.15/Dockerfile.gpu
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:10.0-base-ubuntu18.04
+FROM public.ecr.aws/e2s1w5p1/nvidia-cuda-base:10.0-base-ubuntu18.04
 
 LABEL maintainer="Amazon AI"
 # Specify LABEL for inference pipelines to use SAGEMAKER_BIND_TO_PORT
diff --git a/docker/2.1/Dockerfile.gpu b/docker/2.1/Dockerfile.gpu
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:10.1-base-ubuntu18.04
+FROM public.ecr.aws/e2s1w5p1/nvidia-cuda-base:10.1-base-ubuntu18.04
 
 LABEL maintainer="Amazon AI"
 LABEL com.amazonaws.sagemaker.capabilities.accept-bind-to-port=true

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-FROM nvidia/cuda:10.0-base-ubuntu18.04`
	`1`	`+FROM public.ecr.aws/e2s1w5p1/nvidia-cuda-base:10.0-base-ubuntu18.04`
`2`	`2`
`3`	`3`	`LABEL maintainer="Amazon AI"`
`4`	`4`	`# Specify LABEL for inference pipelines to use SAGEMAKER_BIND_TO_PORT`
Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-FROM nvidia/cuda:10.1-base-ubuntu18.04`
	`1`	`+FROM public.ecr.aws/e2s1w5p1/nvidia-cuda-base:10.1-base-ubuntu18.04`
`2`	`2`
`3`	`3`	`LABEL maintainer="Amazon AI"`
`4`	`4`	`LABEL com.amazonaws.sagemaker.capabilities.accept-bind-to-port=true`