Compartilhar via

Configurar o MKL no Linux

  • Artigo

O padrão CNTK biblioteca matemática é a Biblioteca intel math kernel (Intel MKL). CNTK dá suporte ao uso do Intel MKL por meio de uma versão de biblioteca personalizada MKLML, bem como MKL-DNN neste repositório

Instalando a biblioteca MKL-DNN e MKLML:

  • Crie um diretório em seu computador para armazenar MKLML, por exemplo:
sudo mkdir /usr/local/mklml
  • Baixe o MKLML v0.14 necessário no site do MKLML. Desmarque-o em seu caminho MKLML, criando um subdiretório com versão dentro dele. Criar MKL-DNN usando MKLML e instalar mkl-dnn em /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

Observação: se você quiser compilar com versões MKLML diferentes,

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