Skip to content

Commit 9982855

Browse files
committed
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]>
1 parent 78636eb commit 9982855

File tree

2 files changed

+10
-3
lines changed

2 files changed

+10
-3
lines changed

docker/README.md

+9-2
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,16 @@
1+
# Building a Torch-TensorRT container
12

2-
## Use the container for pytorch1.10+cuda11.1+trt8.0.3.4
3+
### Install Docker and NVIDIA Container Toolkit
4+
5+
https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html
6+
7+
### Build Container
8+
9+
> From root of Torch-TensorRT repo
310
411
```
512
# Build:
6-
docker build -f docker/Dockerfile -t torch_tensorrt1.0:latest .
13+
docker build --build-arg BASE={PyTorch Base Container Version} -f docker/Dockerfile -t torch_tensorrt1.0:latest .
714
815
# Run:
916
docker run --gpus all -it \

docker/dist-build.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
mkdir -p dist
44

5-
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
5+
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
66

77
cd ..
88

0 commit comments

Comments
 (0)