From 0b37ff0557a9b755b62680854d8b12b8af35ee72 Mon Sep 17 00:00:00 2001
From: Sharon Yu <yusharon@amazon.com>
Date: Wed, 14 Aug 2024 12:19:22 -0400
Subject: [PATCH] Upgrade base image

---
 docker/1.7-1/base/Dockerfile.cpu | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/docker/1.7-1/base/Dockerfile.cpu b/docker/1.7-1/base/Dockerfile.cpu
index 6ce800c4..fcd08b4c 100644
--- a/docker/1.7-1/base/Dockerfile.cpu
+++ b/docker/1.7-1/base/Dockerfile.cpu
@@ -1,6 +1,6 @@
 ARG UBUNTU_VERSION=20.04
-ARG CUDA_VERSION=11.6.1
-ARG IMAGE_DIGEST=c2d95c9c6ff77da41cf0f2f9e8c5088f5b4db20c16a7566b808762f05b9032ef
+ARG CUDA_VERSION=12.6.0
+ARG IMAGE_DIGEST=b0937b96f04e1f73279ead20a1d3bb38fb569071756ce1cbde337616beb63129
 
 FROM nvidia/cuda:${CUDA_VERSION}-base-ubuntu${UBUNTU_VERSION}@sha256:${IMAGE_DIGEST}