Created
September 22, 2021 11:34
-
-
Save heiybb/c082645cabb9f0573a1083cee613bc3e to your computer and use it in GitHub Desktop.
tensorrtx yolov5
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
cmake_minimum_required(VERSION 2.6) | |
project(yolov5) | |
set(OpenCV_DIR "D:/IDE/opencv-cuda-build-install") | |
set(TRT_DIR "D:/IDE/CUDA11.2/Dev") | |
add_definitions(-std=c++11) | |
add_definitions(-DAPI_EXPORTS) | |
option(CUDA_USE_STATIC_CUDA_RUNTIME OFF) | |
set(CMAKE_CXX_STANDARD 11) | |
set(CMAKE_BUILD_TYPE Debug) | |
set(THREADS_PREFER_PTHREAD_FLAG ON) | |
find_package(Threads) | |
# setup CUDA | |
find_package(CUDA REQUIRED) | |
message(STATUS " libraries: ${CUDA_LIBRARIES}") | |
message(STATUS " include path: ${CUDA_INCLUDE_DIRS}") | |
set(CUDA_NVCC_PLAGS ${CUDA_NVCC_PLAGS};-std=c++11;-g;-G;-gencode;arch=compute_86;code=sm_86) | |
include_directories(${CUDA_INCLUDE_DIRS}) | |
#### | |
enable_language(CUDA) | |
#### | |
include_directories(${PROJECT_SOURCE_DIR}/include) | |
include_directories(${TRT_DIR}\\include) | |
# -D_MWAITXINTRIN_H_INCLUDED for solving error: identifier "__builtin_ia32_mwaitx" is undefined | |
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11 -Wall -Ofast -D_MWAITXINTRIN_H_INCLUDED") | |
# setup opencv | |
find_package(OpenCV) | |
message(STATUS "OpenCV library status:") | |
message(STATUS " version: ${OpenCV_VERSION}") | |
message(STATUS " libraries: ${OpenCV_LIBS}") | |
message(STATUS " include path: ${OpenCV_INCLUDE_DIRS}") | |
include_directories(${OpenCV_INCLUDE_DIRS}) | |
link_directories(${TRT_DIR}\\lib) | |
add_executable(yolov5 ${PROJECT_SOURCE_DIR}/yolov5.cpp ${PROJECT_SOURCE_DIR}/yololayer.cu ${PROJECT_SOURCE_DIR}/yololayer.h) | |
target_link_libraries(yolov5 "nvinfer" "nvinfer_plugin") | |
target_link_libraries(yolov5 ${OpenCV_LIBS}) | |
target_link_libraries(yolov5 ${CUDA_LIBRARIES}) | |
target_link_libraries(yolov5 Threads::Threads) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment