g

Scheaven

2021-07-26 5016c648cdc5b143845558bca1f9b5bf13acd912

g

 CMakeLists.txt

@@ -23,7 +23,7 @@
set(LIBS nvinfer nvparsers nvonnxparser nvinfer_plugin stdc++fs jsoncpp log4cplus)

if(NOT DEFINED CMAKE_CUDA_ARCHITECTURES)
    set(CMAKE_CUDA_ARCHITECTURES 61)
    set(CMAKE_CUDA_ARCHITECTURES 61 75 80)
endif(NOT DEFINED CMAKE_CUDA_ARCHITECTURES)


@@ -37,6 +37,7 @@
          -gencode arch=compute_61,code=sm_61 -std=c++14# 不同GPU有不同的算力指数，可查看算力表
          -gencode arch=compute_80,code=sm_80 -std=c++14# 不同GPU有不同的算力指数，可查看算力表
          -gencode arch=compute_72,code=sm_72 -std=c++14# 不同GPU有不同的算力指数，可查看算力表
          -gencode arch=compute_75,code=sm_75 -std=c++14# 不同GPU有不同的算力指数，可查看算力表
  )
  set(CUDA_NVCC_FLAGS_RELWITHDEBINFO "--device-debug;-lineinfo")
  #find_package(OpenCV REQUIRED) # 查找系统的默认opencv环境