updated Docker build script

dusty-nv · May 18, 2022 · bed39a4 · bed39a4
1 parent 6cf3b12
commit bed39a4
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/docker/build.sh b/docker/build.sh
@@ -16,7 +16,9 @@ BASE_IMAGE=$1
 source tools/l4t-version.sh
 
 if [ -z $BASE_IMAGE ]; then
-	if [ $L4T_VERSION = "32.6.1" ]; then
+	if [ $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"
 	elif [ $L4T_VERSION = "32.5.1" ]; then
 		BASE_IMAGE="nvcr.io/nvidia/l4t-pytorch:r32.5.0-pth1.6-py3"
@@ -29,8 +31,8 @@ if [ -z $BASE_IMAGE ]; then
 	elif [ $L4T_VERSION = "32.4.2" ]; then
 		BASE_IMAGE="nvcr.io/nvidia/l4t-pytorch:r32.4.2-pth1.5-py3"
 	else
-		echo "cannot build jetson-inference docker container for L4T R$L4T_VERSION"
-		echo "please upgrade to the latest JetPack, or build jetson-inference natively"
+		echo "cannot automatically select l4t-pytorch base container for L4T R$L4T_VERSION"
+		echo "please specify it manually as:  docker/build.sh nvcr.io/nvidia/l4t-pytorch:<TAG>"
 		exit 1
 	fi
 fi