From 8ec4482e499e8a6a82989a185fd7eb879ce21b2d Mon Sep 17 00:00:00 2001 From: Mark Saroufim Date: Wed, 12 Oct 2022 17:48:42 -0700 Subject: [PATCH 1/2] Add correct pip install instructions --- README.md | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/README.md b/README.md index 52232dddb4..1dae40b727 100644 --- a/README.md +++ b/README.md @@ -122,6 +122,12 @@ These are the following dependencies used to verify the testcases. Torch-TensorR Releases: https://github.com/pytorch/TensorRT/releases +``` +pip install nvidia-pyindex +pip install nvidia-tensorrt==8.4.3.1 +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 From 05b9ba2a93f8e7e789bd43cf4f381f7cd21a2a83 Mon Sep 17 00:00:00 2001 From: Mark Saroufim Date: Thu, 13 Oct 2022 13:01:19 -0700 Subject: [PATCH 2/2] add dependencies to toml --- README.md | 2 -- pyproject.toml | 2 ++ 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 1dae40b727..3db7e1035f 100644 --- a/README.md +++ b/README.md @@ -123,8 +123,6 @@ These are the following dependencies used to verify the testcases. Torch-TensorR Releases: https://github.com/pytorch/TensorRT/releases ``` -pip install nvidia-pyindex -pip install nvidia-tensorrt==8.4.3.1 pip install torch-tensorrt==1.2.0 --find-links https://github.com/pytorch/TensorRT/releases/expanded_assets/v1.2.0 ``` 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