在 GPU 上计算距离度量的库

数据挖掘 机器学习 显卡
2022-03-01 07:21:42

广泛使用且快速的 GPU 库的建议,这些库从向量矩阵(相同维度)计算向量的欧几里得距离?

最好是在 NVIDIA GPU 上工作的。

需要它作为比较评估径向基函数神经网络的解决方案的一部分。

1个回答

如果您使用的是 python,请使用带有 cudamat 库的 CUDA 工具包。您将能够使用 cudamat 在 GPU 上执行矩阵运算。