TensorRT Construction
给出一个最简单的创建并运行的过程。
TensorRT Installation
阴差阳错,偶然在 Jetson Xavier 上安装好了 pytorch,torch2trt。值得记录一下。
- 按照 Numba 指南安装,见此处
主要是安装 Numba 会把和 cuda 有关的库都安装好,包括 TensorRT。
- 下载轮子,此处。
- 找到对应的 Torchvision 仓库克隆下来,源码安装。
- 找到
torch2trt安装。
Procedure
- Create a global object
ILogger
ILogger 是个抽象类,需要派生后使用。官方 API 中给了例子。具体使用可以从 Code Sample 中拷贝。
- Create ob

最低0.47元/天 解锁文章
7889

被折叠的 条评论
为什么被折叠?



