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

Question

I have a Makefile containing the following lines, which fish around for the root of the local CUDA tools folder and stuff an environment variable for later consumption. On my system it now yields '/usr/local/cuda-11.4', which is used to find headers and other things.

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

My heart's desire is to replicate this in a CMakeLists.txt file that I am presently building. I want to get the same information in a variable (or something,), and later be able to stuff it into path names. etc.

How can I do this? I have used add_definitions() in simpler ways.

Answer 1

A direct translation of your code would be to use find_program instead of which ,

find_program(NVCC_EXECUTABLE nvcc HINT ENV PATH NO_DEFAULT_PATH)

cmake_path with PARENT_PATH instead of dir

cmake_path(GET ${NVCC_EXECUTABLE} PARENT_PATH CUDA_BIN_PATH)

and string with REPLACE instead of subst .

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

However you could instead use CMake's built-in CUDA support:

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