当前位置:   article > 正文

windows 下 pycuda 安装问题_pycuda安装

pycuda安装

ERROR: Failed building wheel for pycuda
Failed to build pycuda
ERROR: Could not build wheels for pycuda which use PEP 517 and cannot be installed directly

        以上是本人安装出现的问题。


安装不上的主要问题:pip默认安装最新版,导致pycuda版本和系统版本不对应。

解决办法:

1.先查看cuda版本

nvidia-smi

2.查看需要安装的虚拟环境的python版本(如python3.7)

3.前往下载python扩展包(非官网,大牛搞的):https://www.lfd.uci.edu/~gohlke/pythonlibs/#pycudahttps://www.lfd.uci.edu/~gohlke/pythonlibs/#pycuda

找到pycuda 下载对应版本

官方: pycuda · PyPI

浅找了一下,没发现这种类似的方式 

 

 4.下载完成后

到下载的目录下执行安装

例如:

 pip install pycuda-2021.1+cuda114-cp37-cp37m-win_amd64.whl

搞定!!

参考:【GPU加速】安装pycuda异常:Failed to build pycuda ERROR: Could not build wheels for pycuda, which is requir_量化Mike的博客-CSDN博客

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小惠珠哦/article/detail/947010
推荐阅读
相关标签
  

闽ICP备14008679号