Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

onnx->trt 模型转换时,出现显存不够 #27

Open
jiaerwang0328 opened this issue Jun 6, 2024 · 1 comment
Open

onnx->trt 模型转换时,出现显存不够 #27

jiaerwang0328 opened this issue Jun 6, 2024 · 1 comment

Comments

@jiaerwang0328
Copy link

小模型 rtmo-s_8xb32-600e_coco-640x640.onnx 转换时,出现:
Tactic Device request: 9000MB Available: 7982MB. Device memory is insufficient to use tactic
为什么需要这么大的显存?

@Melody-Zhou
Copy link
Owner

小模型 rtmo-s_8xb32-600e_coco-640x640.onnx 转换时,出现: Tactic Device request: 9000MB Available: 7982MB. Device memory is insufficient to use tactic 为什么需要这么大的显存?

maxBatchSize 给小点试试,如下所示:

#! /usr/bin/bash

TRTEXEC=/home/jarvis/lean/TensorRT-8.6.1.6/bin/trtexec

${TRTEXEC} --onnx=rtmo-s_8xb32-600e_coco-640x640.onnx --minShapes=images:1x3x640x640 --optShapes=images:1x3x640x640 --maxShapes=images:4x3x640x640 --saveEngine=rtmo-s_8xb32-600e_coco-640x640.FP32.trtmodel

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants