一、TensorRT环境安装
经过多次踩坑,无论你是Windows还是Linux,或者Ubuntu系统,下面的方法都可以完美安装。
1.1 官网下载
去官网下载,自己系统相对应的压缩包,这个一定要与自己的硬件环境相适配。
链接:NVIDIA TensorRT 8.x Download | NVIDIA Developer
我测试的是8.5 GA的版本,下载需要邮箱登录,自行注册就好。
注意:如果你的是ubuntu系统也是下载linux的就可
1.2 TensorRT安装
若使用的是Anaconda,直接切换到自己的环境,查看python的版本。
python -V
解压下载的压缩包,进入到python文件夹里
找到对应的python版本,cp后面是python的版本(38就是python3.8), 再运行
pip install tensorrt-8.5.1.7-cp38-none-win_amd64.whl
将TensorRT加入到环境变量中, 系统变量PATH
1.3 TensorRT安装验证
import tensorrt as trt
# 尝试创建一个TensorR