From ef2d9743c297d97e328031d821bdce7b5295a6e7 Mon Sep 17 00:00:00 2001 From: maedtb Date: Sat, 3 Aug 2024 11:57:45 -0400 Subject: [PATCH] Update requirements.txt (#56) Add nvidia pip index to fix missing tensorrt_bindings package. Installing `tensorrt` no longer pulls in the required `tensorrt_bindings` package. Adding nvidia's pypi index resolves this issue. Current: `python -s -m pip install "tensorrt>=10.0.1" --dry-run --force-reinstall` ``` Collecting tensorrt>=10.0.1 Using cached tensorrt-10.2.0.post1-py2.py3-none-any.whl Collecting tensorrt-cu12==10.2.0.post1 (from tensorrt>=10.0.1) Using cached tensorrt_cu12-10.2.0.post1-py2.py3-none-any.whl Would install tensorrt-10.2.0.post1 tensorrt-cu12-10.2.0.post1 ``` New: `python -s -m pip install "tensorrt>=10.0.1" --dry-run --force-reinstall --extra-index-url https://pypi.nvidia.com` ``` Looking in indexes: https://pypi.org/simple, https://pypi.nvidia.com Collecting tensorrt>=10.0.1 Using cached https://pypi.nvidia.com/tensorrt/tensorrt-10.2.0.post1.tar.gz (16 kB) Installing build dependencies ... done Getting requirements to build wheel ... done Preparing metadata (pyproject.toml) ... done Collecting tensorrt-cu12==10.2.0.post1 (from tensorrt>=10.0.1) Using cached https://pypi.nvidia.com/tensorrt-cu12/tensorrt-cu12-10.2.0.post1.tar.gz (18 kB) Installing build dependencies ... done Getting requirements to build wheel ... done Installing backend dependencies ... done Preparing metadata (pyproject.toml) ... done Collecting tensorrt-cu12-libs==10.2.0.post1 (from tensorrt-cu12==10.2.0.post1->tensorrt>=10.0.1) Using cached https://pypi.nvidia.com/tensorrt-cu12-libs/tensorrt_cu12_libs-10.2.0.post1-py2.py3-none-manylinux_2_17_x86_64.whl (2017.5 MB) Collecting tensorrt-cu12-bindings==10.2.0.post1 (from tensorrt-cu12==10.2.0.post1->tensorrt>=10.0.1) Using cached https://pypi.nvidia.com/tensorrt-cu12-bindings/tensorrt_cu12_bindings-10.2.0.post1-cp310-none-manylinux_2_17_x86_64.whl (1.1 MB) Collecting nvidia-cuda-runtime-cu12 (from tensorrt-cu12-libs==10.2.0.post1->tensorrt-cu12==10.2.0.post1->tensorrt>=10.0.1) Using cached https://pypi.nvidia.com/nvidia-cuda-runtime-cu12/nvidia_cuda_runtime_cu12-12.6.37-py3-none-manylinux2014_x86_64.whl (897 kB) Would install nvidia-cuda-runtime-cu12-12.6.37 tensorrt-10.2.0.post1 tensorrt-cu12-10.2.0.post1 tensorrt-cu12-bindings-10.2.0.post1 tensorrt-cu12-libs-10.2.0.post1 ``` --- requirements.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index ef4bbb6..bf8c653 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,2 +1,3 @@ tensorrt>=10.0.1 -onnx \ No newline at end of file +onnx +--extra-index-url https://pypi.nvidia.com