-- Check for working CUDA compiler: /usr/bin/nvcc -- broken

当用cmake编译的时候，出现了下面的错误：-- Check for working CUDA compiler: /usr/bin/nvcc -- broken
，我的电脑cuda和cudnn都是安装好了的，出现上面错误的原因可能是编译器找不到cuda或者sm_20和nvcc不兼容，解决办法，在CMakeLists.txt文件最上面添加下面代码：

if (NOT CMAKE_CUDA_COMPILER)
    set(CMAKE_CUDA_COMPILER "/usr/local/cuda-9.0/bin/nvcc")
endif()

参考链接：https://gitlab.kitware.com/cmake/cmake/issues/19396