使用LibTorch C++编译程序时无法找到CUDA的问题求助

可能原因 #

1. Void Linux下CUDA的标准路径软链接未正确指向目标版本，导致CMake检测路径失效
2. 环境变量未被CMake进程正确继承，或CMake 3.26的CUDA检测逻辑需要显式指定路径
3. CMakeLists.txt中手动配置CUDA检测与LibTorch自动处理逻辑冲突

解决步骤 #

1. 修复CUDA路径软链接 #

先检查/usr/local/cuda软链接是否指向正确版本：

ls -l /usr/local/cuda

若输出未指向cuda-12.1，重新创建软链接：

sudo ln -sf /usr/local/cuda-12.1 /usr/local/cuda

2. 显式给CMake传递CUDA路径 #

执行CMake时直接指定CUDA根目录，绕过环境变量检测：

cmake -DCMAKE_PREFIX_PATH=/path/to/your/libtorch -DCUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda-12.1 ..

也可在CMakeLists.txt开头添加强制路径设置：

set(CUDA_TOOLKIT_ROOT_DIR "/usr/local/cuda-12.1")

3. 确保环境变量生效 #

若环境变量是临时设置的，需写入shell配置文件（如~/.bashrc或~/.zshrc），然后执行：

source ~/.bashrc

再重新运行CMake。也可在执行前临时导出变量：

export CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda-12.1 && cmake -DCMAKE_PREFIX_PATH=/path/to/your/libtorch ..

4. 修正CMakeLists.txt配置 #

LibTorch的find_package(Torch)会自动处理CUDA依赖，不要手动添加find_package(CUDA REQUIRED)，避免冲突。标准配置示例：

cmake_minimum_required(VERSION 3.26)
project(example)

set(CMAKE_CXX_STANDARD 17)
set(CMAKE_PREFIX_PATH "/path/to/your/libtorch")

find_package(Torch REQUIRED)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${TORCH_CXX_FLAGS}")

add_executable(example main.cpp)
target_link_libraries(example "${TORCH_LIBRARIES}")
set_property(TARGET example PROPERTY CXX_STANDARD 17)

5. 单独验证CUDA检测 #

创建测试用CMakeLists.txt（空目录下）：

cmake_minimum_required(VERSION 3.26)
project(cuda_test)

find_package(CUDA REQUIRED)
message(STATUS "CUDA版本: ${CUDA_VERSION}")
message(STATUS "CUDA根目录: ${CUDA_TOOLKIT_ROOT_DIR}")
message(STATUS "CUDA头文件路径: ${CUDA_INCLUDE_DIRS}")
message(STATUS "CUDA cudart库: ${CUDA_CUDART_LIBRARY}")

执行检测：

cmake -DCUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda-12.1 .

若此处检测成功，说明问题出在LibTorch配置；若仍报错，需重新安装Void Linux的CUDA包或检查文件权限。

内容的提问来源于stack exchange，提问作者DirtyV