diff --git a/scripts/docker_base.sh b/scripts/docker_base.sh index df7152cb0..65c84cd84 100755 --- a/scripts/docker_base.sh +++ b/scripts/docker_base.sh @@ -19,6 +19,10 @@ if [ $ARCH = "aarch64" ]; then elif [ $L4T_REVISION_MINOR -gt 2 ]; then BASE_IMAGE=$BASE_DEVEL fi + elif [ $L4T_REVISION_MAJOR -eq 7 ]; then + if [ $L4T_REVISION_MINOR -ne 1 ]; then + BASE_IMAGE="nvcr.io/nvidia/l4t-base:r32.7.1" + fi elif [ $L4T_REVISION_MAJOR -gt 7 ]; then BASE_IMAGE=$BASE_DEVEL fi @@ -49,4 +53,4 @@ elif [ $ARCH = "x86_64" ]; then BASE_IMAGE="nvcr.io/nvidia/pytorch:22.04-py3" fi -echo "L4T Base Image: $BASE_IMAGE" \ No newline at end of file +echo "L4T Base Image: $BASE_IMAGE"