dlunion/tensorRTIntegrate

集成libtorch出错

Opened this issue · 1 comments

bigpo commented

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 

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

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