法一:安装 pycuda
a, pip3 install pycuda==版本
b, 根据这个 链接 https://www.lfd.uci.edu/~gohlke/pythonlibs/#pycuda,主要根据你的python 版本确定,没有对应好pycuda 和 python, 会出现各种问题.
ps:注意conda 环境和和本地环境 python,默认会找本地环境,可用 …/…/envs/…/bin/pip3 安装到conda 虚拟环境
法二:(推荐)
直接pip安装不上的主要问题:pip默认安装最新版,导致pycuda版本和系统版本不对应。
解决办法:
1.先查看cuda版本
如何查看可看我这篇博客:如何查看自己的cuda环境,以及其对应版本_请让我按时毕业的博客-优快云博客
2. 查看当前虚拟环境的python版本
python -V
3. 下载对应版本的pycuda库
pycuda库下载地址:https://www.lfd.uci.edu/~gohlke/pythonlibs/#pycuda
4. 进行pycuda库安装
进入下载pycuda库所在的目录下进入CMD环境:
pip install pycuda所在位置