首先 ,查看你的~/.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
我是重启后有效的