使用vs2019编译 c++ libtorch 的 yolov5项目出现的问题,尤其是 Microsoft C++ 异常: c10::Error,位于内存位置 0x0000000D606FE350

 

 

使用vs2019编译 c++ libtorch 的 yolov5项目出现的问题,尤其是 Microsoft C++ 异常: c10::Error,位于内存位置 0x0000000D606FE350_第1张图片

inline c10::intrusive_ptr IValue::toTuple() const & {
  AT_ASSERT(isTuple(), "Expected Tuple but got ", tagKind());
  return toIntrusivePtr();
}
 

这里先说说我是怎么导出.pt模型的,yolov5官方提供了导出的python的代码,在如下:

问题就出现这个转换的代码里,作者说,这个转换默认是调用cpu 的,所以如何使用GPU需要重新修改一下:

你需要修改几句:

 

github作者的回复:https://github.com/yasenh/libtorch-yolov5#torchscript-model-export

你可能感兴趣的:(libtorch部署,c++,pytorch)