| | |
|---|
| | | source_group("Source" FILES ${sources}) |
|---|
| | | |
|---|
| | | |
|---|
| | | #cudaS |
|---|
| | | find_package(CUDA REQUIRED) |
|---|
| | | |
|---|
| | | #message("-- CUDA version:$(CUDA_VERSION)") |
|---|
| | | link_directories(/usr/local/cuda/lib64) |
|---|
| | | link_directories(/usr/local/lib) |
|---|
| | | |
|---|
| | | # 设置环境的模型查找，如果CUDA版本大于9,则执行CUDA环境的模型查找 |
|---|
| | | if(CUDA_VERSION_MAJOR GREATER 9) |
|---|
| | |
|---|
| | | endif() |
|---|
| | | |
|---|
| | | |
|---|
| | | set(OpenCV_DIR /data/disk2/opt/01_opencv/opencv4.5.1/build) |
|---|
| | | FIND_PACKAGE(OpenCV REQUIRED) |
|---|
| | | |
|---|
| | | link_directories(/usr/local/cuda-10.0/lib64) |
|---|
| | | #link_directories(/usr/local/lib) |
|---|
| | | # set(OpenCV_DIR /data/disk2/opt/01_opencv/opencv4.5.1/build) |
|---|
| | | FIND_PACKAGE(OpenCV REQUIRED PATHS "/data/disk2/opt/01_opencv/opencv4.5.2") |
|---|
| | | include_directories( ${OpenCV_INCLUDE_DIRS} ) |
|---|
| | | |
|---|
| | | SET(LIBS |
|---|
| | | gomp |
|---|
| | |
|---|
| | | #include_directories("/usr/include/json/") |
|---|
| | | #include_directories("/usr/include") |
|---|
| | | #include_directories("/usr/local/include/") |
|---|
| | | include_directories(${OPENCV_DIR}/include) |
|---|
| | | |
|---|
| | | |
|---|
| | | #设置链接库搜索目录，项目中lib路径 |
|---|
| | | link_directories(${PYTORCH_DIR}/lib) #动态链接库目录/bazel-bin/tensorflow |
|---|
