如何在 numpy Python 中启用和禁用 Intel MKL？

Question

我想测试和比较使用英特尔 MKL 和不使用英特尔 MKL 的 Numpy 矩阵乘法和特征分解性能。

我已经使用pip install mkl（Windows 10（64位），Python 3.8）安装了MKL。

然后我使用这里的示例进行 matmul 和特征分解。

我现在如何启用和禁用 MKL 以便检查使用 MKL 和不使用 MKL 时的 numpy 性能？

参考代码：

import numpy as np
from time import time

def matrix_mul(size, n=100):
    # reference: https://markus-beuckelmann.de/blog/boosting-numpy-blas.html
    np.random.seed(112)
    a, b = np.random.random((size, size)), np.random.random((size, size))
    t = time()
    for _ in range(n):
        np.dot(a, b)
    delta = time() - t
    print('Dotted two matrices of size %dx%d in %0.4f ms.' % (size, size, delta / n * 1000))


def eigen_decomposition(size, n=10):
    np.random.seed(112)
    a = np.random.random((size, size))
    t = time()
    for _ in range(n):
        np.linalg.eig(a)
    delta = time() - t
    print('Eigen decomposition of size %dx%d in %0.4f ms.' % (size, size, delta / n * 1000))

#Obtaining computation times: 

for i in range(20): 
    eigen_decomposition(500)

for i in range(20): 
    matrix_mul(500)

Answer 1

您可以使用不同的环境来比较带有和不带有 MKL 的 Numpy。在每个环境中，您可以使用软件包安装程序安装所需的软件包（带有或不带有 MKL 的 numpy）。然后，在该环境中，您可以运行程序来比较使用和不使用 MKL 的 Numpy 的性能。

\n

NumPy 不\xe2\x80\x99 不依赖于任何其他 Python 包，但是，它确实依赖于加速线性代数库 - 通常是 Intel MKL 或 OpenBLAS。

\n

\n

• PyPI 上的 NumPy 轮（即 pip 安装的）是使用 OpenBLAS 构建的。

  \n
\n

• 在 conda 默认通道中，NumPy 是针对 Intel MKL 构建的。MKL 是一个单独的软件包，将在用户安装 NumPy 时安装在用户的环境中。

  \n
\n

• 当用户从 conda-forge 安装 NumPy 时，该 BLAS 包会与实际库一起安装。但它也可以是 MKL（来自默认通道），甚至是 BLIS 或参考 BLAS。

  \n
\n

\n

请参考此链接详细了解如何安装 Numpy。

\n

您可以创建两个不同的环境来比较使用 MKL 和不使用 MKL 时的 NumPy 性能。在第一个环境中安装独立的 NumPy（即不带 MKL 的 NumPy），在第二个环境中安装带 MKL 的 NumPy。

\n

使用 NumPy 而不使用 MKL 创建环境。

\n

conda create -n <env_name_1> python=<version>\nconda activate <env_name_1>\npip install numpy\n

\n

但根据您的操作系统，可能没有可用的发行版 (Windows)。

\n

在 Windows 上，我们一直都是针对 MKL 进行链接。然而，在 Anaconda 2.5 版本中，我们将 MKL 运行​​时分离到自己的 conda 包中，以便在所有平台上统一执行操作。

\n

一般来说，你可以创建一个新的环境：

\n

conda create -n wheel_based python\nactivate wheel\npip install numpy-1.13.3-cp36-none-win_amd64.whl  # or whatever the file is named\n

\n

在其他环境中，使用以下命令安装带有 MKL 的 NumPy

\n

conda create -n <env_name_2> python=<version>\nconda activate <env_name_2>\npip install intel-numpy\n

\n

在这些环境 <env_name_1> 和 <env_name_2> 中，您可以单独运行您的程序，以便您可以分别比较没有 MKL 和有 MKL 的 Numpy 的性能。

\n