直接pip install pycuda出错了
出错的原因是找不到cuda.h这个文件,将cuda路径添加到环境变量即可:
找到cuda的路径,我这里是 cuda-10.2
在命令行输入以下命令就可以了:
sudo gedit ~/.bashrc
export PATH=/usr/local/cuda-10.2/bin:$PATH
source ~/.bashrc
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64
pip install pycuda --user -i https://pypi.tuna.tsinghua.edu.cn/simple