File tree 2 files changed +10
-3
lines changed
2 files changed +10
-3
lines changed Original file line number Diff line number Diff line change
1
+ # Building a Torch-TensorRT container
1
2
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
3
10
4
11
```
5
12
# 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 .
7
14
8
15
# Run:
9
16
docker run --gpus all -it \
Original file line number Diff line number Diff line change 2
2
3
3
mkdir -p dist
4
4
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
6
6
7
7
cd ..
8
8
You can’t perform that action at this time.
0 commit comments