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 ..