安装pycuda及问题解决

  直接通过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

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容