updates for JetPack 5.1.2 / L4T R35.4.1

dusty-nv · Aug 30, 2023 · 44c7661 · 44c7661
1 parent 3d0d0f8
commit 44c7661
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/docker/build.sh b/docker/build.sh
@@ -17,7 +17,11 @@ source docker/tag.sh
 
 if [ -z $BASE_IMAGE ]; then
 	if [ $ARCH = "aarch64" ]; then
-		if [ $L4T_VERSION = "35.2.1" ]; then
+		if [ $L4T_VERSION = "35.4.1" ]; then
+			BASE_IMAGE="dustynv/l4t-pytorch:r35.4.1"
+		elif [ $L4T_VERSION = "35.3.1" ]; then
+			BASE_IMAGE="dustynv/l4t-pytorch:r35.3.1"
+		elif [ $L4T_VERSION = "35.2.1" ]; then
 			BASE_IMAGE="nvcr.io/nvidia/l4t-pytorch:r35.2.1-pth2.0-py3"
 		elif [ $L4T_VERSION = "35.1.0" ]; then
 			BASE_IMAGE="nvcr.io/nvidia/l4t-pytorch:r35.1.0-pth1.12-py3"

diff --git a/docker/tag.sh b/docker/tag.sh
@@ -28,8 +28,8 @@ if [ $ARCH = "aarch64" ]; then
 			CONTAINER_TAG="r32.7.1"
 		fi
 	elif [ $L4T_RELEASE -eq 35 ]; then
-		if [ $L4T_REVISION_MAJOR -gt 3 ]; then
-			CONTAINER_TAG="r35.3.1"
+		if [ $L4T_REVISION_MAJOR -gt 4 ]; then
+			CONTAINER_TAG="r35.4.1"
 		fi
 	fi