pip install tensorrt
时间: 2025-01-23 19:11:22 浏览: 69
### 如何使用 `pip` 安装 TensorRT 库
为了成功安装 TensorRT 库,建议先确保环境中的 `pip` 和其他依赖项是最新的:
```bash
python3 -m pip install --upgrade setuptools pip
```
接着,在 Linux 下可以按照如下方式通过 `pip` 来安装 TensorRT 及其相关索引服务:
```bash
pip install nvidia-pyindex
pip install nvidia-tensorrt
```
需要注意的是,并不是所有的 TensorRT 版本都支持直接通过 `pip` 进行安装。对于特定版本如 9.3.0 的安装可能会遇到困难[^3]。
如果希望简化安装过程并减少潜在错误,可以选择安装经过优化后的精简版或调度版 TensorRT 组件包:
#### 安装精简版 TensorRT
适用于仅需运行预构建的兼容版本引擎的情况:
```bash
pip install tensorrt-lean-cu11 -i https://pypi.ngc.nvidia.com
```
#### 安装调度版 TensorRT
提供更全面的功能集:
```bash
pip install tensorrt-dispatch-cu11 -i https://pypi.ngc.nvidia.com
```
此外,当计划利用 ONNX 模型时,可能还需要额外安装 ONNX 或者 ONNX Runtime 支持:
```bash
pip install onnx
pip install onnxruntime
# 如果有GPU需求则考虑gpu版本
pip install onnxruntime-gpu
```
阅读全文
相关推荐



















