CMAKE_CUDA_ARCHITECTURES如何设置

CMAKE_CUDA_ARCHITECTURES 用于指定编译 CUDA 代码时所需的 GPU 架构。可以在 CMakeLists.txt 文件中设置此选项。

例如，要将 CMAKE_CUDA_ARCHITECTURES 设置为 sm_60 和 sm_61，可以在 CMakeLists.txt 文件中添加以下行：

set(CMAKE_CUDA_ARCHITECTURES 60 61)

这将告诉 CMake 编译器需要支持 sm_60 和 sm_61 架构的 GPU。

另外，CMAKE_CUDA_ARCHITECTURES 还支持多个 GPU 架构，可以使用分号分隔它们。例如：

set(CMAKE_CUDA_ARCHITECTURES 70;75;80)

这将告诉 CMake 编译器需要支持 sm_70、sm_75 和 sm_80 架构的 GPU。