https://github.com/Abraham423/CenterPointTensorRT 的cmake
2023-09-14 09:09:26 时间
cmake_minimum_required(VERSION 2.8.3)
project(centerpoint)
set(USE_CUDA True)
# For TensorRT sample lib
# set(TRT_ROOT /home/wanghao/Desktop/projects/TensorRT)
include_directories(../TensorRT-8/lib)
include_directories(../TensorRT-8/include)
include_directories(
common
)
set(SAMPLES_COMMON_SOURCES
common/logger.cpp
)
#set(SAMPLE_SOURCES samplecenterpoint.cpp)
#set(SAMPLE_PARSERS "onnx")
#include(
#~/TensorRT/samples/CMakeSamplesTemplate.txt
# )
# pcl and boost related !
#find_package(PCL REQUIRED)
#find_package(Boost COMPONENTS program_options REQUIRED )
#include_directories(${Boost_INCLUDE_DIRS})
#link_directories(${Boost_LIBRARY_DIRS})
message("cut Boost")
#include_directories(${PCL_INCLUDE_DIRS})
#link_directories(${PCL_LIBRARY_DIRS})
#add_definitions(${PCL_DEFINITIONS})
message("cut pcl")
# set flags for CUDA availability
option(CUDA_AVAIL "CUDA available" OFF)
find_package(CUDA)
message("CUDA dir paths", ${CUDA_LIBRARIES})
include_directories(
${PROJECT_SOURCE_DIR}
${PROJECT_SOURCE_DIR}/include
${CUDA_INCLUDE_DIRS}
)
message("cut tensorrt..")
file(GLOB PointPillarLibs ${PROJECT_SOURCE_DIR}/**cpp)
add_library(pointpillars SHARED ${PointPillarLibs})
file(GLOB PointPillarCU ${PROJECT_SOURCE_DIR}/**cu)
cuda_add_library(pointpillarsCU SHARED ${PointPillarCU})
set(CUDA_LIB_PATH /usr/local/cuda/lib64/)
add_executable(centerpoint samplecenterpoint.cpp ${SAMPLES_COMMON_SOURCES})
target_link_libraries(pointpillars nvonnxparser)
target_link_libraries(centerpoint
#${PCL_LIBRARIES}
#${Boost_LIBRARIES}
${CUDA_LIBRARY}
${CUDA_RUNTIME_LIBRARY}
nvonnxparser
nvinfer
pointpillars
${CUDA_LIB_PATH}libcudart.so
pointpillarsCU
)
#add_executable(samplerpn samplerpn.cpp ${SAMPLES_COMMON_SOURCES})
#target_link_libraries(samplerpn
# ${PCL_LIBRARIES}
# ${Boost_LIBRARIES}
# ${CUDA_LIBRARY}
# ${CUDA_RUNTIME_LIBRARY}
# nvonnxparser
# nvinfer
# pointpillars
# ${CUDA_LIB_PATH}libcudart.so
# #pointpillarsCU
# )
#file(GLOB PointPillarLibs ${PROJECT_SOURCE_DIR}/nodes/**)
##file(GLOB PointPillarCU ${PROJECT_SOURCE_DIR}/nodes/**cu)
#
#message("PointPillarLibsNodes",${PointPillarLibs})
##add_library(point_pillars SHARED ${PointPillarLibs})
#cuda_add_library(point_pillars SHARED ${PointPillarLibs})
#
##link_directories(${PROJECT_SOURCE_DIR}/nodes)
#link_libraries (
# point_pillars
# ${PCL_LIBRARIES}
# ${Boost_LIBRARIES}
# ${CUDA_LIBRARY}
# ${CUDA_RUNTIME_LIBRARY}
# )
#add_executable(lidar_point_pillars ${PROJECT_SOURCE_DIR}/nodes/lidar_point_pillar_test.cpp)
#
#target_link_libraries (lidar_point_pillars
# ${PCL_LIBRARIES}
# ${Boost_LIBRARIES}
# ${CUDA_LIBRARY}
# ${CUDA_RUNTIME_LIBRARY}
# point_pillars
# nvonnxparser
# )
相关文章
- GitHub 近两万Star,无需编码,可一键生成前后端代码,这个开源项目JeecgBoot有点强
- Github-Dorks与辅助工具
- 天池移动推荐算法赛--https://github.com/PnYuan/Tianchi-BigData
- 如何参与一个GitHub开源项目
- Android开源项目分包方式学习(eoe、oschina、github)
- github page搭建自己的页面
- [FAQ] GitHub 开启二次验证之后,如何通过 https clone 项目 ?
- GitHub 推出了集成了 GPT-4 技术的 Copilot X,它对程序员有哪些积极和消极影响?
- github仓库上的漏洞修复
- 使用经过oauth验证后的github API,避免调用频次超标的问题
- GitHub上那些值得一试的Java开源库
- Github上如何找到自己想要的开源项目(小技巧:精确搜索)
- github帐户和仓库的创建
- GitHub上传文件夹及Git命令
- github代码外泄监控——可用来提供源码泄露检测服务,数据泄露场景,原理就是在github搜索关键字
- 【Mac 系统】GitHub从创建Repo到上传项目源码详细步骤
- fatal: unable to access ‘https://github.com/用户名/项目名.git/‘: OpenSSL SSL_read: Connection was reset, e