要在 Python 中安装 TensorRT 8.6 版本，你需要按照以下几个步骤操作。请注意，TensorRT 主要支持 Linux 和 Docker 环境，在 Windows 上的支持有限。以下是详细的安装指南： ### 安装准备 1. **确认硬件兼容性** - 确保你的 GPU 支持 CUDA 并且驱动程序已更新至最新版本。 2. **安装 NVIDIA 驱动和 CUDA Toolkit** - 下载并安装适用于你系统的 [CUDA Toolkit](https://developer.nvidia.com/cuda-downloads) 。建议选择与你要安装的 TensorRT 版本相匹配的 CUDA 版本。 3. **安装 cuDNN** - 下载并安装 [cuDNN](https://developer.nvidia.com/rdp/form/cudnn-download-survey) ，这是 CUDA 的深度神经网络库，对于大多数深度学习框架都是必需的。 4. **设置环境变量** - 将 CUDA 和 cuDNN 添加到系统的 PATH 和 LD_LIBRARY_PATH 变量中。 ### 安装 TensorRT 8.6 5. **下载 TensorRT 安装包** - 访问 [NVIDIA 开发者网站](https://developer.nvidia.com/nvidia-tensorrt-8x)，注册账号并登录后可以找到 TensorRT 8.x 的发行版。 - 根据操作系统选择合适的 tar 文件或 deb 包进行下载（例如 `TensorRT-8.6.0.xxx.Linux.x86_64-gnu.cuda-xx.cudnn-x.xx.tar.gz` 或 `.deb`）。 6. **解压文件并安装** ```bash # 解压缩 tar 文件 tar zxvf TensorRT-8.6.0.xxx.Linux.x86_64-gnu.cuda-xx.cudnn-x.xx.tar.gz # 移动解压后的文件夹内容到指定位置 sudo cp -r TensorRT-8.6.0 /usr/local/ # 更新动态链接库缓存 sudo ldconfig /usr/local/TensorRT-8.6.0/lib ``` 7. **通过 pip 安装 Python 接口** - 如果你想直接从 PyPI 获取预编译的轮子文件（whl），可以尝试使用下面命令： ```bash pip install nvidia-tensorrt==8.6.* ``` - 但是需要注意的是官方不一定提供所有平台及配置下的 whl 文件；如果没有适合的 wheel，则需要手动构建 Python 绑定部分。 8. **验证安装成功与否** - 运行简单的测试脚本来确保一切正常工作： ```python import tensorrt as trt print(trt.__version__) ``` 如果上述过程顺利完成并且打印出了正确的版本信息 (`8.6.x`)，那么就说明安装已经成功！