一、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
# 尝试创建一个tensorrt的logger
logger
发表评论