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
搞定!!