MIUAV
MIUAV/vibe-coding-ros2
用 Codex 或 Claude 帮你安装 复制这段 Prompt，粘贴到 Codex、Claude 或其他助手里，让它检查 Skill 页面并帮你完成安装。
直接命令不会经过审查 Prompt；运行前请先检查来源。
npx skills add https://github.com/MIUAV/vibe-coding-ros2 --skill tensorrt命令会保持在同一行。复制前请横向滚动并检查完整内容。
想先保存到本地？可下载 SkillsMP 当前能够提供的文件。
正在显示 SKILL.md
基于 SOC 职业分类
| name | tensorrt |
| description | TensorRT推理优化 - 模型转换 引擎构建 INT8量化 |
| argument-hint | TensorRT OR 模型优化 OR INT8量化 OR 引擎生成 |
| user-invocable | true |
NVIDIA推理优化引擎
ONNX/PyTorch -> TensorRT -> Engine -> Inference
import tensorrt as trt
logger = trt.Logger(trt.Logger.WARNING)
builder = trt.Builder(logger)
network = builder.create_network(1 << int(trt.NetworkDefinitionCreationFlag.EXPLICIT_BATCH))
config = builder.create_builder_config()
parser = trt.OnnxParser(network, logger)
# 解析ONNX
with open('model.onnx', 'rb') as f:
parser.parse(f.read())
# 构建引擎
engine = builder.build_serialized_network(network, config)
# 推理
runtime = trv.Runtime(logger)
engine = runtime.deserialize_cuda_engine(engine)
context = engine.create_execution_context()
context.execute_v2(bindings)