如何从cmake（3.10版本）将CUDA架构设置为compute_50和sm_50？

3 回答

• 2

  所以我能够自己解决这个问题 . 我们可以通过以下方式设置它 -

  string（APPEND CMAKE_CUDA_FLAGS“-gencode arch = compute_50，code = sm_50”）

  回复于 2024-05-09T10:17:46+08:00
• 0

  正确的方法是：

  target_compile_options(myTarget PRIVATE $<$<COMPILE_LANGUAGE:CUDA>:-gencode arch=compute_50,code=sm_50>)
  

  根据需要选择PRIVATE / PUBLIC . 这是设置每个目标标志的正确方法 .

  回复于 2024-05-09T10:17:46+08:00
• 2

  我written a module默认获取（和使用）第一个已安装的物理设备的计算机功能设置，但可以通过设置CUDA_SM环境变量来覆盖它 . 这是模块（将内容放在 .cmake 文件中并将其放在包含的文件夹中）：

  # This module determines which compute capability / SM version
  # we should be compiling our CUDA code for, and adds the appropriate
  # switch to the NVCC compiler flags - so that you don't have to worry
  # about it.
  #
  # TODO: Be willing to take CUDA_CC, CUDA_TARGET_COMPUTE_CAPABILITY, 
  # CUDA_TARGET_COMPUTE or CUDA_TARGET_COMPUTE_CAP and maybe even 
  # those without the CUDA_ prefix
  
  if (NOT CUDA_TARGET_COMPUTE_CAPABILITY)
          if("$ENV{CUDA_SM}" STREQUAL "")
                  set(ENV{CUDA_INCLUDE_DIRS} "${CUDA_INCLUDE_DIRS}")
                  set(ENV{CUDA_CUDART_LIBRARY} "${CUDA_CUDART_LIBRARY}")
                  set(ENV{CMAKE_CXX_COMPILER} "${CMAKE_CXX_COMPILER}")
                  execute_process(COMMAND bash -c "${CMAKE_CURRENT_SOURCE_DIR}/scripts/get_cuda_sm.sh"  OUTPUT_VARIABLE CUDA_TARGET_COMPUTE_CAPABILITY_) 
          else()
                  set(CUDA_TARGET_COMPUTE_CAPABILITY_ $ENV{CUDA_SM})
          endif()
  
          set(CUDA_TARGET_COMPUTE_CAPABILITY "${CUDA_TARGET_COMPUTE_CAPABILITY_}" CACHE STRING "CUDA compute capability of the (first) CUDA device on the system, in XY format (like the X.Y format but no dot); see table of features and capabilities by capability X.Y value at https://en.wikipedia.org/wiki/CUDA#Version_features_and_specifications")
  
          execute_process(COMMAND bash -c "echo -n $(echo ${CUDA_TARGET_COMPUTE_CAPABILITY})" OUTPUT_VARIABLE CUDA_TARGET_COMPUTE_CAPABILITY) 
          execute_process(COMMAND bash -c "echo ${CUDA_TARGET_COMPUTE_CAPABILITY} | sed 's/^\\([0-9]\\)\\([0-9]\\)/\\1.\\2/;' | xargs echo -n" OUTPUT_VARIABLE FORMATTED_COMPUTE_CAPABILITY) 
  
          message(STATUS "CUDA device-side code will assume compute capability ${FORMATTED_COMPUTE_CAPABILITY}")
  endif()
  
  set(CUDA_GENCODE "arch=compute_${CUDA_TARGET_COMPUTE_CAPABILITY},code=compu
  

  回复于 2024-05-09T10:17:46+08:00