search

dlunion / tensorRTIntegrate

TensorRT ONNX Plugin、Inference、Compile
463 stars 114 forks source link

集成libtorch出错 #26

Open bigpo opened 4 years ago

bigpo commented 4 years ago

hi, @dlunion @hopef 感谢这个工作，我目前想在代码里面集成libtorch，但是在ubuntu下链接到主函数就报错，不知是我写的问题，还是bug？

你们有尝试集成libtorch吗？

我在Makefile里加入的有关torch是这样的部分

INC_TORCH := $(TORCH_PREFIX)/include $(TORCH_PREFIX)/include/torch/csrc/api/include

LIB_TORCH := $(TORCH_PREFIX)/lib /usr/local/cuda/lib64/stubs

LD_TORCH := torch c10 c10d c10_cuda

不知道可否帮忙看下问题在哪儿？ 十分感谢！

dlunion commented 4 years ago

我不确定你报错的具体内容，可否携带具体错误信息？