From 69bb7370a5ce747faa3e22aff3f1896aa8a8d802 Mon Sep 17 00:00:00 2001 From: Francesco Gatti Date: Wed, 30 Mar 2022 15:52:30 +0200 Subject: [PATCH] compile with tensorrt7 --- CMakeLists.txt | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index 1af836d..13db7b5 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -78,8 +78,7 @@ endif() #------------------------------------------------------------------------------- # CUDA #------------------------------------------------------------------------------- - -#set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} -arch=sm_30 --compiler-options '-fPIC'") +set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS}" --compiler-options '-fPIC') find_package(CUDNN REQUIRED)