强制numpy使用OpenBlas

时间:2019-06-09 16:15:14

标签: python numpy openblas

基本上,我需要将两个(10 ^ 6 x 10 ^ 6)整数元素数组相乘,而我无法使用numpy.dot。 查看其他帖子,我发现如果将矩阵的类型切换为float32或更大的值,并且安装了openblas,则点运算将更快。 实际上,我将数组tipe更改为float32,实际上执行起来要快得多。

所以我的问题是,假设numpy确实在使用OpenBlas,是否有任何方法可以强制numpy使用OpenBlas库,这样我仍然可以将矩阵作为int类型?

如果需要帮助,我在这里给我我的Numpy配置。

np.__config__.show()
blas_mkl_info:
  NOT AVAILABLE
blis_info:
  NOT AVAILABLE
openblas_info:
    libraries = ['openblas', 'openblas']
    library_dirs = ['/usr/local/lib']
    language = c
    define_macros = [('HAVE_CBLAS', None)]
blas_opt_info:
    libraries = ['openblas', 'openblas']
    library_dirs = ['/usr/local/lib']
    language = c
    define_macros = [('HAVE_CBLAS', None)]
lapack_mkl_info:
  NOT AVAILABLE
openblas_lapack_info:
    libraries = ['openblas', 'openblas']
    library_dirs = ['/usr/local/lib']
    language = c
    define_macros = [('HAVE_CBLAS', None)]
lapack_opt_info:
    libraries = ['openblas', 'openblas']
    library_dirs = ['/usr/local/lib']
    language = c
    define_macros = [('HAVE_CBLAS', None)]

0 个答案:

没有答案
相关问题