A high performance deep learning inference library
Project description
NVIDIA TensorRT is an SDK that facilitates high-performance machine learning inference. It is designed to work in a complementary fashion with training frameworks such as TensorFlow, PyTorch, and MXNet. It focuses specifically on running an already-trained network quickly and efficiently on NVIDIA hardware.
IMPORTANT: This is a special release of TensorRT designed to work only with TensorRT-LLM. Please refrain from upgrading to this version if you are not using TensorRT-LLM.
To install, please execute the following:
pip install tensorrt --extra-index-url https://pypi.nvidia.com
Or add the index URL to the (space-separated) PIP_EXTRA_INDEX_URL environment variable:
export PIP_EXTRA_INDEX_URL='https://pypi.nvidia.com'
pip install tensorrt
When the extra index url does not contain https://pypi.nvidia.com
, a nested pip install
will run with the proper extra index url hard-coded.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Hashes for tensorrt_lean-cu12-10.0.0b6.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | a5f33ece323ffa9836b711f82821e1f54bbeb374619513d2d4251c36decbe3bd |
|
MD5 | 1ca01a2d42189623be0f41ac5dab23e0 |
|
BLAKE2b-256 | a6bf07fd7f0b14ed2f21188f3d62f802a79ba7971e895df2b071b5228dc3c5ca |