赞
踩
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
搞定!!
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。