一、进入官方链接下载对应版本的tensorRT
NVIDIA TensorRT 8.x Download | NVIDIA Developer
1.进入以上链接后找到这个页面
2.下载与自己cuda对应版本的tensorRT
EA 版本代表抢先体验,GA 是稳定版。建议使用GA版本
二、下载好后移动相应文件夹
1.将下载好的tensorRT文件夹中的include文件移动到已下载好的CUDA文件夹下的include文件夹
2.将下载好的tensorRT文件夹中的lib中的dll文件放到已下载好的CUDA下的bin文件夹
3.将下载好的tensorRT文件夹中的lib中的lib文件放到已下载好的CUDA下的lib\x64文件夹
三、pip安装文件夹内的whl文件
一共需要安装4个whl文件,在如图所示的红框中
我是在虚拟环境下安装的,所以在虚拟环境下打开终端
pip install +要安装的whl文件的绝对路径
这三个选项唯一直接安装就好,下面这个根据自己的python版本选择
四、检测安装是否成功
在安装环境里运行代码成功则安装成功
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import tensorrt as trt
if __name__ == "__main__":
print(trt.__version__)
print("hello trt!!")
如果出现报错说缺了dll文件,就在下面这个链接下载所缺失的dll文件,放入报错的文件夹