From 7ea5161b59e40f8b05620d53c9bbf7c7cbdff206 Mon Sep 17 00:00:00 2001
From: Dustin Franklin <dustinf@nvidia.com>
Date: Fri, 23 Oct 2020 12:14:50 -0400
Subject: [PATCH] updated Docker build for L4T R32.4.4

---
 docker/build.sh | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/docker/build.sh b/docker/build.sh
index 44f0ad260..f09767408 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.4.3" ]; then
+	if [ $L4T_VERSION = "32.4.4" ]; then
+		BASE_IMAGE="nvcr.io/nvidia/l4t-pytorch:r32.4.4-pth1.6-py3"
+	elif [ $L4T_VERSION = "32.4.3" ]; then
 		BASE_IMAGE="nvcr.io/nvidia/l4t-pytorch:r32.4.3-pth1.6-py3"
 	elif [ $L4T_VERSION = "32.4.2" ]; then
 		BASE_IMAGE="nvcr.io/nvidia/l4t-pytorch:r32.4.2-pth1.5-py3"