zl程序教程

您现在的位置是:首页 >  后端

当前栏目

Python GPU 加速数据科学 | 计算距离矩阵在用 cuPy 时快了约 100 倍

Python计算数据 矩阵 加速 100 GPU 科学
2023-09-27 14:19:44 时间

CSDN 叶庭云https://yetingyun.blog.csdn.net/


一、环境配置

利用 cuPy 的 cupyx.scipy.spatial.distance 方法计算距离矩阵时,这个 distance 模块使用 pylibraft 作为后端,因此还需要安装好 pylibraft package 才行,可以直接从 Conda 安装,链接为:https://anaconda.org/rapidsai/pylibraft

也可以用 pip install 安装 pylibraft-cu11cupy-cuda11x(注意:我本地 CUDA 版本为 11.3,因此选择 pylibraft-cu11 与 cupy-cuda