Ubuntu 下在安装部分安装包,例如 Pycuda 时可能会遇到这个问题:
src/cpp/cuda.hpp:14:18: fatal error: cuda.h: No such file or directory
下面以安装 pycuda 为例:
解决方法:
sudo su
export PATH=/usr/local/cuda-9.0/bin:/usr/local/cuda/bin:$PATH // 即 cuda 的路径这里在root 模式下再包含一遍即可
pip install pycuda