gpt4 book ai didi

Numpy、BLAS 和 CUBLAS

转载 作者:行者123 更新时间:2023-12-03 14:50:38 24 4
gpt4 key购买 nike

Numpy 可以针对不同的 BLAS 实现(MKL、ACML、ATLAS、GotoBlas 等)“链接/编译”。这并不总是很容易配置,但它是可能的。

是否也可以针对 NVIDIA 的 CUBLAS 实现“链接/编译”numpy?
我在网络上找不到任何资源,在我花太多时间尝试之前,我想确保它完全可行。

最佳答案

一句话:不,你不能那样做。

有一个相当不错的 scikit,它提供从 scipy 访问 CUBLAS 的权限,称为 scikits.cuda 。它建立在 PyCUDA 之上. PyCUDA 提供了一个 numpy.ndarray类似于允许使用 CUDA 在 GPU 内存中无缝操作 numpy 数组的类。所以你可以将 CUBLAS 和 CUDA 与 numpy 一起使用,但你不能只链接到 CUBLAS 并期望它能够工作。

还有一个商业库,它提供类似 numpy 和 cublas 的功能,并且具有 Python 接口(interface)或绑定(bind),但我将把它留给他们的一个工作人员来填补这一点。

关于Numpy、BLAS 和 CUBLAS,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/11576073/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com