Merge branch 'master' of https://github.com/dusty-nv/jetson-inference

shoaib42 · Aug 9, 2021 · 5601ec6 · 5601ec6
2 parents f27cb53 + 5cacc8a
commit 5601ec6
Showing 1 changed file with 3 additions and 1 deletion.
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.5.1" ]; then
+	if [ $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"
 	elif [ $L4T_VERSION = "32.5.0" ]; then
 		BASE_IMAGE="nvcr.io/nvidia/l4t-pytorch:r32.5.0-pth1.6-py3"