问题描述

– The CUDA compiler identification is unknown
CMake Error at /usr/share/cmake-3.24/Modules/CMakeDetermineCUDACompiler.cmake:602 (message):
Failed to detect a default CUDA architecture.

在正确安装CUDA之后 cmake时输出“CUDA compiler identification is unknown”以及“Failed to detect a default CUDA architecture.”

查阅相关资料说明是CMAKE版本的问题

Compiler output:

在这里插入图片描述

解决方案

cmake时加入指定参数来给参数初始化
查找自己的GPU所对应的Architecture赋值CMAKE

查看网址:https://developer.nvidia.com/cuda-gpus

添加变量标签

在这里插入图片描述

-DCMAKE_CUDA_ARCHITECTURES=86
-DCMAKE_CUDA_COMPILER=“/usr/local/cuda/bin/nvcc”

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐