tensorRT安装

tensorRT对于生产环境中的实时和低延迟需求非常关键

TensorRT通过层融合、优化内存使用、量化以及自动搜索最优执行计划等多种手段,极大提升了模型在部署阶段的执行效率.

寻找对应cuda版本的tensorRT下载
https://developer.nvidia.com/tensorrt/download

tensorRT与cuda的版本对应表
https://blog.csdn.net/weixin_41540237/article/details/131589929

解压后创建环境变量“D:\xxx\TensorRT-8.6.1.6\lib”

若要使用python:

#对应的文件夹和whl包,不一定和这个一莫一样
cd TensorRT-5.0.2.6/python
pip install tensorrt-5.0.2.6-py2.py3-none-any.whl

验证:

import tensorrt

报错:

FileNotFoundError: Could not find: nvinfer.dll. Is it on your PATH?
Note: Paths searched were:
['C:\\ProgramData\\anaconda3\\Scripts', 'C:\\Prog

解决:
先配置环境变量再创建虚拟环境,或重启pycharm

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值