通过

你当前正在访问 Microsoft Azure Global Edition 技术文档网站。 如果需要访问由世纪互联运营的 Microsoft Azure 中国技术文档网站，请访问 https://docs.azure.cn。

在 Linux 上设置 MKL

  • 项目

默认CNTK数学库是 Intel 数学内核库 (Intel MKL) 。 CNTK支持通过自定义库版本 MKLML 以及此存储库中的 MKL-DNN 使用 Intel MKL

安装 MKL-DNN 和 MKLML 库：

  • 在计算机上创建一个目录以保存 MKLML，例如：
sudo mkdir /usr/local/mklml
  • 从 MKLML 网站下载所需的 MKLML v0.14。 将其解压缩到 MKLML 路径中，在其中创建版本控制子目录。 使用 MKLML 生成 MKL-DNN，并将 mkl-dnn 安装到 /usr/local
sudo wget https://github.com/01org/mkl-dnn/releases/download/v0.14/mklml_lnx_2018.0.3.20180406.tgz
sudo tar -xzf mklml_lnx_2018.0.3.20180406.tgz -C /usr/local/mklml
wget --no-verbose -O - https://github.com/01org/mkl-dnn/archive/v0.14.tar.gz | tar -xzf - && \
cd mkl-dnn-0.14 && \
ln -s /usr/local external && \
mkdir -p build && \
cd build && \
cmake .. && \
make && \
sudo make install && \
cd ../.. && \
rm -rf mkl-dnn-0.14

注意：如果要使用不同的 MKLML 版本进行生成，

  • 在) 下方配置生成 (cf 时，请指定选项 --with-mkl=<directory>，例如 --with-mkl=/usr/local/mkl/mklml_lnx_2018.0.3.20180406。