diff --git a/README.md b/README.md index 52232dddb4..3db7e1035f 100644 --- a/README.md +++ b/README.md @@ -122,6 +122,10 @@ These are the following dependencies used to verify the testcases. Torch-TensorR Releases: https://github.com/pytorch/TensorRT/releases +``` +pip install torch-tensorrt==1.2.0 --find-links https://github.com/pytorch/TensorRT/releases/expanded_assets/v1.2.0 +``` + ## Compiling Torch-TensorRT ### Installing Dependencies diff --git a/pyproject.toml b/pyproject.toml index 38456cf53d..bb07e2f7cf 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -9,6 +9,8 @@ requires = [ "cffi", "typing_extensions", "future", + "nvidia-pyindex", + "nvidia-tensorrt==8.4.3.1" ] # Use legacy backend to import local packages in setup.py