直接通过pip安装pycuda报错:
ERROR: Could not build wheels for pycuda, which is required to install pyproject.toml-based projects
原因:pip默认安装最新版,导致pycuda版本和系统版本不对应。
解决方案:
(1)查看系统cuda版本,比如:cuda10.1
(2)查看当前虚拟环境的python版本,不如:python3.7
(3)前往https://www.lfd.uci.edu/~gohlke/pythonlibs/#pycuda网站,根据cuda版本和python版本查找对应的pycuda版本。比如,这里的应该是:pycuda‑2020.1。
pycuda版本解释:
如:pycuda‑2020.1+cuda101‑cp37‑cp37m‑win_amd64.whl
pycuda‑2020.1表示pycuda版本为 2020.1
cuda101表示对应cuda版本为10.1
cp37表示Python解释器版本为3.7
找到对应的版本即可,这里可以不下载,用pip命令下载安装
(原文链接:https://blog.csdn.net/qq_41910905/article/details/109650182)