From 9b311caaf3c84e33258125b6d036db3134380e84 Mon Sep 17 00:00:00 2001
From: Dylan Auty <dylan.auty@gmail.com>
Date: Wed, 19 Jul 2023 14:28:11 +0100
Subject: [PATCH] Point to r32.7.1 base image for all L4T 32.7.* versions

---
 scripts/docker_base.sh | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/scripts/docker_base.sh b/scripts/docker_base.sh
index df7152cb0..65c84cd84 100755
--- a/scripts/docker_base.sh
+++ b/scripts/docker_base.sh
@@ -19,6 +19,10 @@ if [ $ARCH = "aarch64" ]; then
 			elif [ $L4T_REVISION_MINOR -gt 2 ]; then
 				BASE_IMAGE=$BASE_DEVEL
 			fi
+		elif [ $L4T_REVISION_MAJOR -eq 7 ]; then
+			if [ $L4T_REVISION_MINOR -ne 1 ]; then
+				BASE_IMAGE="nvcr.io/nvidia/l4t-base:r32.7.1"
+			fi
 		elif [ $L4T_REVISION_MAJOR -gt 7 ]; then
 			BASE_IMAGE=$BASE_DEVEL
 		fi
@@ -49,4 +53,4 @@ elif [ $ARCH = "x86_64" ]; then
 	BASE_IMAGE="nvcr.io/nvidia/pytorch:22.04-py3"
 fi
 
-echo "L4T Base Image:   $BASE_IMAGE"
\ No newline at end of file
+echo "L4T Base Image:   $BASE_IMAGE"