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 75 80) set(CMAKE_CUDA_ARCHITECTURES 61 75 72 80) endif(NOT DEFINED CMAKE_CUDA_ARCHITECTURES)