From bed39a4ed9f6477ac7337295bf6713a46caebaec Mon Sep 17 00:00:00 2001 From: Dustin Franklin Date: Wed, 18 May 2022 11:04:11 -0400 Subject: [PATCH] updated Docker build script --- docker/build.sh | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/docker/build.sh b/docker/build.sh index 140a9abb8..94ac47387 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.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:" exit 1 fi fi