Partilhar via

Configuração MKL em Linux

  • Artigo

A biblioteca de matemática CNTK padrão é a Biblioteca Intel Math Kernel (Intel MKL). CNTK suporta usar o Intel MKL através de uma versão de biblioteca personalizada MKLML, bem como MKL-DNN neste repo

Instalação da biblioteca MKL-DNN e MKLML:

  • Crie um diretório na sua máquina para segurar a MKLML, por exemplo:
sudo mkdir /usr/local/mklml
  • Descarregue o MKLML v0.14 necessário a partir do site da MKLML. Desatar-o no seu caminho MKLML, criando um sub-diretório em versão dentro. Construa a MKL-DNN utilizando a MKLML e instale a mkl-dnn para /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

Nota: se quiser construir com diferentes versões MKLML,

  • Ao configurar a construção (cf. abaixo), especifique a opção --with-mkl=<directory>, por exemplo, --with-mkl=/usr/local/mkl/mklml_lnx_2018.0.3.20180406.