共用方式為

在 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 。