Switch base image to NVIDIA's CUDA base

Signed-off-by: GitHub <noreply@github.com>
ministryofjustice · May 16, 2024 · e67763d · e67763d
1 parent 7a80ff3
commit e67763d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -1,4 +1,4 @@
-FROM public.ecr.aws/ubuntu/ubuntu@sha256:1cc4019c277242e5fb710f2cf40af1590ad053b7d764efba48b051871dcc18d8
+FROM docker.io/nvidia/cuda:12.4.1-base-ubuntu22.04
 
 LABEL org.opencontainers.image.vendor="Ministry of Justice" \
       org.opencontainers.image.authors="Analytical Platform (analytical-platform@digital.justice.gov.uk)" \