Partager via

Configurer MKL sur Linux

  • Article

La bibliothèque mathématique par défaut CNTK est la bibliothèque Intel Math Kernel Library (Intel MKL). CNTK prend en charge l’utilisation d’Intel MKL via une version de bibliothèque personnalisée MKLML, ainsi que MKL-DNN dans ce référentiel

Installation de la bibliothèque MKL-DNN et MKLML :

  • Créez un répertoire sur votre ordinateur pour contenir MKLML, par exemple :
sudo mkdir /usr/local/mklml
  • Téléchargez le fichier MKLML v0.14 requis à partir du site web MKLML. Supprimez-la dans votre chemin MKLML, en créant un sous-répertoire avec version. Générer MKL-DNN à l’aide de MKLML et installer mkl-dnn sur /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

Remarque : si vous souhaitez générer avec différentes versions MKLML,

  • Lors de la configuration de la build (cf. ci-dessous), spécifiez l’option --with-mkl=<directory>, par exemple --with-mkl=/usr/local/mkl/mklml_lnx_2018.0.3.20180406.