From 998285531bf598f8be9ff84ed5a61388f7f9ad47 Mon Sep 17 00:00:00 2001
From: Naren Dasan <naren@narendasan.com>
Date: Mon, 25 Oct 2021 12:06:20 -0400
Subject: [PATCH] fix(//docker): Update docker container build script to use
 release path to not be dependent on git

Signed-off-by: Naren Dasan <narens@nvidia.com>
Signed-off-by: Naren Dasan <naren@narendasan.com>
---
 docker/README.md     | 11 +++++++++--
 docker/dist-build.sh |  2 +-
 2 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/docker/README.md b/docker/README.md
index f3ee0aa151..d1a6fb6ac0 100644
--- a/docker/README.md
+++ b/docker/README.md
@@ -1,9 +1,16 @@
+# Building a Torch-TensorRT container
 
-## Use the container for pytorch1.10+cuda11.1+trt8.0.3.4
+### Install Docker and NVIDIA Container Toolkit
+
+https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html
+
+### Build Container
+
+> From root of Torch-TensorRT repo
 
 ```
 # Build:
-docker build -f docker/Dockerfile -t torch_tensorrt1.0:latest .
+docker build --build-arg BASE={PyTorch Base Container Version} -f docker/Dockerfile -t torch_tensorrt1.0:latest .
 
 # Run:
 docker run --gpus all -it \
diff --git a/docker/dist-build.sh b/docker/dist-build.sh
index a76b19a3a3..08d9d5cff9 100755
--- a/docker/dist-build.sh
+++ b/docker/dist-build.sh
@@ -2,7 +2,7 @@
 
 mkdir -p dist
 
-cd py && MAX_JOBS=1 LANG=en_US.UTF-8 LANGUAGE=en_US:en LC_ALL=en_US.UTF-8 python3 setup.py bdist_wheel --use-cxx11-abi
+cd py && MAX_JOBS=1 LANG=en_US.UTF-8 LANGUAGE=en_US:en LC_ALL=en_US.UTF-8 python3 setup.py bdist_wheel --use-cxx11-abi --release
 
 cd ..