TRT_LOGGER Verbosity: Severity.ERROR

举报
风吹稻花香 发表于 2021/06/06 00:05:49 2021/06/06
【摘要】   TRT_LOGGER Verbosity: Severity.ERROR   https://github.com/NVIDIA/TensorRT/issues/289   # Start TensorRT 6 Container nvidia-docker run -it -v ${PWD}:/mnt --workdir=/mnt...

 

TRT_LOGGER Verbosity: Severity.ERROR

 

https://github.com/NVIDIA/TensorRT/issues/289

 

# Start TensorRT 6 Container
nvidia-docker run -it -v ${PWD}:/mnt --workdir=/mnt nvcr.io/nvidia/tensorrt:19.12-py3

# Install dependencies
pip install torch==1.4 torchvision==0.5

# Download test model code
wget https://gist.githubusercontent.com/rmccorm4/b72abac18aed6be4c1725db18eba4930/raw/fbdb009152ef54abe7d8b23a9fd57a0f250e03e4/alexnet_onnx.py

# Create test model
python alexnet_onnx.py --opset=11

# Parse test model
trtexec --onnx=alexnet_fixed.onnx --explicitBatch

...
In node 0 (importModel): INVALID_GRAPH: Assertion failed: tensors.count(input_name)
&&&& FAILED TensorRT.trtexec # trtexec --onnx=alexnet_fixed.onnx --explicitBatch

文章来源: blog.csdn.net,作者:网奇,版权归原作者所有,如需转载,请联系作者。

原文链接:blog.csdn.net/jacke121/article/details/105252127

【版权声明】本文为华为云社区用户转载文章,如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。