我需要一个 CMakeLists.txt 等效于定义环境变量的此 Makefile 行

Question

我有一个 Makefile 包含以下几行，它在本地 CUDA 工具文件夹的根目录中四处寻找，并填充一个环境变量供以后使用。 在我的系统上，它现在生成“/usr/local/cuda-11.4”，用于查找标题和其他内容。

# Location of the CUDA Toolkit
CUDA_PATH = $(subst /bin/,,$(dir $(shell which nvcc)))

我的心愿是在我目前正在构建的 CMakeLists.txt 文件中复制它。 我想在变量（或其他东西）中获得相同的信息，然后能够将其填充到路径名中。 等等

我怎样才能做到这一点？ 我以更简单的方式使用了 add_definitions() 。

Answer 1

您的代码的直接翻译是使用find_program而不是which ，

find_program(NVCC_EXECUTABLE nvcc HINT ENV PATH NO_DEFAULT_PATH)

使用cmake_path而不是dir的PARENT_PATH

cmake_path(GET ${NVCC_EXECUTABLE} PARENT_PATH CUDA_BIN_PATH)

和string用REPLACE而不是subst 。

string(REPLACE /bin/ "" CUDA_PATH ${CUDA_BIN_PATH})

但是，您可以改为使用 CMake 的内置 CUDA 支持：

find_package(CUDAToolkit REQUIRED)
set(CUDA_PATH ${CUDAToolkit_LIBRARY_ROOT})