✅ TensorRT Python 安装精简流程(适用于 Ubuntu 20.04+)

安装 TensorRT Python 轮子的步骤

确保 pipwheel 模块已更新并安装:

参考链接

复制代码
`python3 -m pip install --upgrade pip
python3 -m pip install wheel`

1. 确认环境要求

  • Python:版本 3.8 - 3.13

  • OS:Ubuntu 20.04+ 或 Windows 10+

  • CPU:x86_64 或 ARM SBSA 架构

  • 安装前确保 pip、wheel 是最新的:

    python3 -m pip install --upgrade pip setuptools wheel packaging


2. 清理旧版本(建议操作)

复制代码
pip cache remove "tensorrt*"

3. 安装 TensorRT(默认 CUDA 12.x 版本)

复制代码
python3 -m pip install --upgrade tensorrt

若你需要 指定 CUDA 版本(如 CUDA 11),则:

复制代码
python3 -m pip install tensorrt-cu11 tensorrt-lean-cu11 tensorrt-dispatch-cu11

4. 轻量版本安装(仅推理使用)

复制代码
python3 -m pip install --upgrade tensorrt-lean
python3 -m pip install --upgrade tensorrt-dispatch

5. 没有 root 权限?

加上 --user 参数:

复制代码
python3 -m pip install --user tensorrt

6. 验证安装是否成功

打开 Python,运行以下命令:

🔹 常规版本验证:
复制代码
import tensorrt
print(tensorrt.__version__)
assert tensorrt.Builder(tensorrt.Logger())
🔹 精简版本验证:
复制代码
import tensorrt_lean as trt
print(trt.__version__)
assert trt.Runtime(trt.Logger())

import tensorrt_dispatch as trt
print(trt.__version__)
assert trt.Runtime(trt.Logger())

⚠️ 错误排查

如果报错:

复制代码
[TensorRT] ERROR: CUDA initialization failure with error 100

可能原因:

  • 没有安装 NVIDIA 驱动

  • 驱动与 CUDA 不兼容

  • 在容器中未挂载 GPU

👉 可尝试使用 NVIDIA 官方容器:

复制代码
docker run --gpus all -it --rm nvidia/cuda:12.3.2-base-ubuntu20.04

需要我帮你检查 CUDA 和驱动是否匹配,或者生成一个完整的安装脚本吗?

相关推荐
磨十三1 小时前
C++ 标准库排序算法 std::sort 使用详解
开发语言·c++·排序算法
薰衣草23332 小时前
力扣——位运算
python·算法·leetcode
156082072192 小时前
在飞腾D2000/8平台下ubuntu内核添加WX1860和WX1820的驱动
linux·ubuntu
两只程序猿2 小时前
数据可视化 | Violin Plot小提琴图Python实现 数据分布密度可视化科研图表
开发语言·python·信息可视化
野生技术架构师3 小时前
1000 道 Java 架构师岗面试题
java·开发语言
湫兮之风3 小时前
C++: Lambda表达式详解(从入门到深入)
开发语言·c++
Porunarufu3 小时前
JAVA·顺序逻辑控制
java·开发语言
Sylvia-girl3 小时前
C语言中经常使用的函数
c语言·开发语言
~无忧花开~3 小时前
JavaScript学习笔记(十五):ES6模板字符串使用指南
开发语言·前端·javascript·vue.js·学习·es6·js
周杰伦fans3 小时前
C# 23种设计模式详解与示例
开发语言·设计模式·c#