fix(//docker): Update docker container build script to use release path

to not be dependent on git Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
pytorch · Oct 25, 2021 · 9982855 · 9982855
1 parent 78636eb
commit 9982855
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git 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
@@ -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 ..