haotian/duan8
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added Target_link_libraries for myplugins

Browse Source 
Based on @philipp-schmidt fix on his repository, https://github.com/wang-xinyu/tensorrtx/issues/139#issuecomment-674698203 I've added the link libraries to make sure that the plugin importation works well in Python TensorRT API.

The same will probably have to be done for all other implementations in this repository.
This commit is contained in:
cesarandreslopez 2020-08-17 12:22:15 +04:00 committed by wang-xinyu
parent 5dc79e982e
commit 1199fa23ce
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
yolov5/CMakeLists.txt
View File

@ -36,6 +36,7 @@ target_link_libraries(yolov5 nvinfer)
target_link_libraries(yolov5 cudart)
target_link_libraries(yolov5 myplugins)
target_link_libraries(yolov5 ${OpenCV_LIBS})
target_link_libraries(myplugins nvinfer cudart)
add_definitions(-O2 -pthread)