From 5cacc8a3c755d1d943b1902fc7120da8a1a209f2 Mon Sep 17 00:00:00 2001 From: Dustin Franklin Date: Mon, 9 Aug 2021 14:20:06 -0400 Subject: [PATCH] updated base images for JetPack 4.6 --- docker/build.sh | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/docker/build.sh b/docker/build.sh index 7a07f37c3..d46d8aa52 100755 --- 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"