环境:ubuntu 16.04
可通过以下指令直接安装:
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pycuda --no-binary :all:
如果通过上述命令安装过程出现如下问题:
ERROR: Could not build wheels for pycuda, which is required to install pyproject.toml-based projects
是因为通过 pip 默认安装的都是最新版本,而最新版的 pycuda 可能和 cuda 版本不一致,因为 pycuda 更新速度较慢,只适配了部分版本的 cuda 。
可通过下载对应自己系统 cuda 版本的 pycuda 的.whl 文件,手动安装,下载地址:
https://www.lfd.uci.edu/~gohlke/pythonlibs/#pycudahttps://www.lfd.uci.edu/~gohlke/pythonlibs/#pycuda