windows 下 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博客

猜你喜欢

转载自blog.csdn.net/qq_63019407/article/details/123707287