环境:ubuntu 16.04

可通过以下指令直接安装:

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pycuda --no-binary :all:

如果通过上述命令安装过程出现如下问题:

ERROR: Could not build wheels for pycuda, which is required to install pyproject.toml-based projects

是因为通过 pip 默认安装的都是最新版本,而最新版的 pycuda 可能和 cuda 版本不一致,因为 pycuda 更新速度较慢,只适配了部分版本的 cuda 。

可通过下载对应自己系统 cuda 版本的 pycuda 的.whl 文件,手动安装,下载地址:

https://www.lfd.uci.edu/~gohlke/pythonlibs/#pycudaicon-default.png?t=M1H3https://www.lfd.uci.edu/~gohlke/pythonlibs/#pycuda

Logo

欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐