hongshaorou/YOPO
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add abort when CUDA Arch auto-detection failed

Browse Source
This commit is contained in:
TJU_Lu 2025-06-21 14:52:53 +08:00
parent 491edcafbe
commit 4e09f3b64a
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
Simulator/src/CMakeLists.txt
View File

@ -23,6 +23,13 @@ message(WARNING "OpenCV_VERSION: ${OpenCV_VERSION}")
find_package(CUDA REQUIRED) find_package(CUDA REQUIRED)
# Query the GPU architecture # Query the GPU architecture
cuda_select_nvcc_arch_flags(ARCH_FLAGS) cuda_select_nvcc_arch_flags(ARCH_FLAGS)
if("${ARCH_FLAGS}" STREQUAL "")
message(FATAL_ERROR
"Failed to automatically detect CUDA architecture. "
"Please manually set ARCH_FLAGS in CMakeList.txt (line 25) based on your device, for example:\n"
" set(ARCH_FLAGS \"-gencode arch=compute_86,code=sm_86\")"
)
endif()
message(WARNING "CUDA Architecture: ${ARCH_FLAGS}") message(WARNING "CUDA Architecture: ${ARCH_FLAGS}")
set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} ${ARCH_FLAGS}") set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} ${ARCH_FLAGS}")