在使用CMAKE编译是找不到CUDA,可以添加参数解决,错误如下
The CUDA compiler identification is unknown
CMake Error at /home/XXX_USER/anaconda3/lib/python3.10/site-packages/cmake/data/share/cmake-3.27/Modules/CMakeDetermineCUDACompiler.cmake:603 (message):
Failed to detect a default CUDA architecture.
可以添加cuda安装目录解决:
cmake .. -DUSE_CUDA=ON -D CMAKE_CUDA_COMPILER="/usr/local/cuda/bin/nvcc"
顺利解决问题