whu-wxy / psenet-libtorch Goto Github PK

大佬你好。最近在搞部署这一块。很幸运看到你的这个部署repo。大佬能把模型从pytorch代码转为torchscript完整的代码开源一下吗？小白感觉无从下手。感谢

https://gitee.com/zhou_rx/codes/jpfuc5yd4vgaqm03ti8b785

jit.trace自己训练的模型，总是失败 请问怎么回事？

Hi @Whu-wxy
I wrote a simple program in C++ to use my PyTorch model in libtorch 1.7.1. I have CUDA 10.2 on my OS, thus I downloaded CUDA 10.2 prebuilt version of libtorch from this link :.link

This is my simple program :

This program works fine in CMake , but when i run it in Qt , it gives me this error :

This is my QMake file :

QT -= gui
CONFIG += c++14 console
CONFIG -= app_bundle
DEFINES += QT_DEPRECATED_WARNINGS
SOURCES +=
main.cpp
qnx: target.path = /tmp/$${TARGET}/bin
else: unix:!android: target.path = /opt/$${TARGET}/bin
!isEmpty(target.path): INSTALLS += target
QMAKE_CXXFLAGS += -DGLIBCXX_USE_CXX11_ABI=0
CONFIG += no_keywords
QMAKE_CXXFLAGS += -Wno-unused-parameter
QMAKE_LFLAGS += -INCLUDE:?warp_size@cuda@at@@yahxz
INCLUDEPATH += /home/eaglesoft/libtorch/include/torch/csrc/api/include
INCLUDEPATH += /home/eaglesoft/libtorch/include
DEPENDPATH += /home/eaglesoft/libtorch/include/torch/csrc/api/include
DEPENDPATH += /home/eaglesoft/libtorch/include
LIBS += -L/home/eaglesoft/libtorch/lib -lc10
-lc10_cuda
-lcaffe2_module_test_dynamic
-lcaffe2_observers
-ltorch_cpu
-ltorch
-lc10d_cuda_test
-lcaffe2_detectron_ops_gpu
-lcaffe2_nvrtc
-lfmt
-lnvrtc-builtins
-lshm
-ltensorpipe
-ltorch_cuda

Any help would be appreciated.