法一:安装 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环境,以及其对应版本_请让我按时毕业的博客-CSDN博客

2. 查看当前虚拟环境的python版本

python -V

 

 3. 下载对应版本的pycuda库

pycuda库下载地址:https://www.lfd.uci.edu/~gohlke/pythonlibs/#pycuda

 4. 进行pycuda库安装

 进入下载pycuda库所在的目录下进入CMD环境:

pip install pycuda所在位置 

参考文章:windows 下 pycuda 安装问题_pycuda安装_不喝可乐不快乐的博客-CSDN博客

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐