-
Notifications
You must be signed in to change notification settings - Fork 2.1k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Installing tensorrt 10.0.1 pypi package installs wrong version of tensorrt-cu12 #3945
Comments
Looks like can workaround with |
You're absolutely right, pip is installing TensorRT 10.1 instead of the desired 10.0.1 when using the extra index URL with --extra-index-url https://pypi.nvidia.com. Here's how to address this issue: Understanding the Problem: The NVIDIA PyPI repository seems to prioritize newer versions on the index. Even though you specify ==10.0.1, it might not install that specific version because 10.1 is available. Downgrade After Installation: Install TensorRT 10.1 first: Use a Virtual Environment: Create a virtual environment to isolate TensorRT 10.0.1 and prevent conflicts with other projects. If strict version control is crucial, consider building TensorRT 10.0.1 from source. Download the source code from the NVIDIA NGC catalog and follow the build instructions. For most scenarios, using a virtual environment (solution 2) is the preferred approach. It provides a clean and isolated environment for specific project dependencies. Additional Considerations: Make sure your CUDA Toolkit and NVIDIA Driver versions are compatible with TensorRT 10.0.1. You can find compatibility information in the TensorRT documentation. |
We have filed this bug internally and plan to fix it for the next release. In the meantime, please use |
Description
pip install --extra-index-url https://pypi.nvidia.com tensorrt==10.0.1
now installs tensorrt 10.1 instead of 10.0.1 and fails to deserialize engines built with 10.0.1Steps To Reproduce
It looks like the metapackage tensorrt has
tensorrt-cu12
dependency which is not pinned to a specific version so older version of tensorrt is installing latest version of tensorrt-cu12.The text was updated successfully, but these errors were encountered: