python mkl如何查看

我不是码神2024-04-13前端开发6

MKL（Intel Math Kernel Library，英特尔数学核心库）是英特尔公司开发的一款高性能数学计算库，它为科学计算、工程计算和数据分析提供了丰富的函数接口，在Python中，我们可以使用NumPy、SciPy等科学计算库，这些库底层都依赖于MKL进行数学计算，那么如何查看Python程序是否使用了MKL呢？本文将详细介绍如何在Python环境中查看MKL的使用情况。

（图片来源网络，侵删）

1、我们需要安装Intel的MKL库，在Windows系统中，可以从英特尔官网下载MKL库的安装包，然后按照提示进行安装，在Linux系统中，可以使用包管理器进行安装，例如在Ubuntu系统中，可以使用以下命令安装：

sudo aptget install libmklintellp64dev

2、安装完成后，我们可以使用ldd命令查看Python程序是否加载了MKL库，在终端中输入以下命令：

ldd /usr/bin/python3

如果输出结果中包含libmkl_rt.so或libmkl_core.so等MKL库文件，说明Python程序已经加载了MKL库。

...
libmkl_rt.so => /opt/intel/compilers_and_libraries_2019.5.284/linux/mkl/lib/intel64/libmkl_rt.so (0x00007f8e7c3a7000)
...

3、除了使用ldd命令外，我们还可以使用nm命令查看Python程序中的符号信息，在终端中输入以下命令：

nm D /usr/bin/python3 | grep mkl

如果输出结果中包含__imp_mkl_cdft_*、__imp_mkl_dfti*、__imp_mkl_dcdft*等MKL相关的符号，说明Python程序已经使用了MKL库的相关功能。

...
00000000002bcf90 T __imp_mkl_cdft_execute
...

4、我们还可以通过Python的ctypes库来检查MKL库的版本信息，我们需要找到MKL库的头文件路径，在Linux系统中，通常位于/opt/intel/compilers_and_libraries_<version>/linux/include目录下。

echo "/opt/intel/compilers_and_libraries_2019.5.284/linux/include"

接下来，我们编写一个简单的Python脚本来检查MKL库的版本信息：

import ctypes
from ctypes import cdll, c_char_p, c_void_p, sizeof, byref, cast, c_int, c_longlong, c_ubyte, c_uint, c_double, c_float, c_bool, c_short, c_long, c_ulong, c_ushort, c_char, c_wchar, c_void, c_size_t, c_ssize_t, c_int8, c_uint8, c_int16, c_uint16, c_int32, c_uint32, c_int64, c_uint64, c_complex, c_doublecomplex, c_bool, c_byte, c_ubyte, c_short, c_ushort, c_long, c_ulong, c_longlong, c_ulonglong, c_float, c_double, c_char16, c_char32, c_wchar16, c_wchar32, cdll
设置头文件路径
mkl = cdll.LoadLibrary("/opt/intel/compilers_and_libraries_2019.5.284/linux/include/mkl.h")
获取MKL版本信息
mkl.mklGetVersion(byref(major), byref(minor))
print("MKL version: {}.{}".format(major.value, minor.value))

运行上述脚本，如果输出结果中包含MKL的版本号，说明Python程序已经加载了MKL库并使用了其相关功能。