-- 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