TensorRT推理报错:pycuda._driver.LogicError: cuMemcpyHtoDAsync failed: invalid argument

文章目录:

遇到这个错误的原因一般有两个:

  • tensorrt推理引擎输入推理数据的维度不对
  • tensorrt推理引擎输入推理数据的类型不对

因此,请仔细检查输入数据的类型和维度信息是否正确!!!

参考1:https://blog.csdn.net/GungnirsPledge/article/details/108428651
参考2:https://111qqz.github.io/2020/09/torch2trt/

欢迎大家关注笔者,你的关注是我持续更博的最大动力


原创文章,转载告知,盗版必究

微信:suihailiang0816
QQ:931762054
wx公众号:仰望星空的小随
已标记关键词 清除标记
相关推荐
©️2020 CSDN 皮肤主题: 猿与汪的秘密 设计师:白松林 返回首页