updated container base images

dusty-nv · Aug 11, 2022 · 692fdd1 · 692fdd1
1 parent ffc88ea
commit 692fdd1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/docker/build.sh b/docker/build.sh
@@ -17,7 +17,9 @@ source docker/tag.sh
 
 if [ -z $BASE_IMAGE ]; then
 	if [ $ARCH = "aarch64" ]; then
-		if [ $L4T_VERSION = "34.1.0" ]; then
+		if [ $L4T_VERSION = "34.1.1" ]; then
+			BASE_IMAGE="nvcr.io/nvidia/l4t-pytorch:r34.1.1-pth1.12-py3"
+		elif [ $L4T_VERSION = "34.1.0" ]; then
 			BASE_IMAGE="nvcr.io/nvidia/l4t-pytorch:r34.1.0-pth1.12-py3"
 		elif [ $L4T_VERSION = "32.6.1" ]; then
 			BASE_IMAGE="nvcr.io/nvidia/l4t-pytorch:r32.6.1-pth1.9-py3"