File tree Expand file tree Collapse file tree 2 files changed +10
-3
lines changed Expand file tree Collapse file tree 2 files changed +10
-3
lines changed Original file line number Diff line number Diff line change 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:
916docker run --gpus all -it \
Original file line number Diff line number Diff line change 22
33mkdir -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
77cd ..
88
You can’t perform that action at this time.
0 commit comments