From 5016c648cdc5b143845558bca1f9b5bf13acd912 Mon Sep 17 00:00:00 2001
From: Scheaven <xuepengqiang>
Date: 星期一, 26 七月 2021 17:34:19 +0800
Subject: [PATCH] g
---
CMakeLists.txt | 3 ++-
1 files changed, 2 insertions(+), 1 deletions(-)
diff --git a/CMakeLists.txt b/CMakeLists.txt
index 31b884c..3f92f33 100644
--- a/CMakeLists.txt
+++ b/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) # 鏌ユ壘绯荤粺鐨勯粯璁pencv鐜
--
Gitblit v1.8.0