文章目录
- 前言
- TensorRT下载
- TensorRT安装
- 参考资料
前言
本文将介绍Windows如何安装TensorRT。本文的基础是:Windows安装Pytorch+CUDA环境
TensorRT下载
进入官方网站:https://developer.nvidia.com/nvidia-tensorrt-8x-download
寻找自己对应的版本,我这里选择为:
下载得到 zip 压缩包,解压。
TensorRT安装
任意顺序完成以下几步:
- 复制TensorRT-8.4.3.1\bin中内容到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\bin
- 复制TensorRT的include文件夹到CUDA的include文件夹
- 复制TensorRT-8.4.3.1\lib文件夹中的lib文件到CUDA的lib文件夹,dll文件到CUDA的bin文件夹
- 使用pip install xxx.whl安装TensorRT-8.4.3.1文件夹中的
如下图所示:
使用 python 检查是否安装成功。
参考资料
TensorRT8.4安装教程