Поделиться через

Настройка MKL в Linux

  • Статья

По умолчанию CNTK математической библиотекой является библиотека ядра Intel Math (Intel MKL). CNTK поддерживает использование Intel MKL с помощью пользовательской версии библиотеки MKLML, а также MKL-DNN в этом репозитории.

Установка библиотеки MKL-DNN и MKLML:

  • Создайте каталог на компьютере для хранения MKLML, например:
sudo mkdir /usr/local/mklml
  • Скачайте требуемый MKLML версии 0.14 с веб-сайта MKLML. Распакуйте его в пути MKLML, создав подкаталог с версиями в пределах. Сборка MKL-DNN с помощью MKLML и установка 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.