首先 ,查看你的~/.bashrc 文件 裏面有沒有
export PATH="/usr/local/cuda-9.0/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH"
其次看看你的makefile文件是否對cuda的地址是/usr/local/cuda-9.0 有的默認是/usr/local/cuda,所以要ln -s /usr/local/cuda-9.0 /usr/local/cuda
如果以上情況都不是,並且locate cuda_runtime.h 能找到正確位置
那麼試試這個命令吧
sudo apt-get install nvidia-cuda-toolkit
我是重啓後有效的