Sdílet prostřednictvím


Nastavení MKL v Linuxu

Výchozí CNTK matematická knihovna je Intel Math Kernel Library (Intel MKL). CNTK podporuje použití Intel MKL prostřednictvím vlastní verze knihovny MKLML a MKL-DNN v tomto úložišti.

Instalace knihovny MKL-DNN a MKLML:

  • Vytvořte na svém počítači adresář pro uchovávání MKLML, například:
sudo mkdir /usr/local/mklml
  • Stáhněte si požadovaný MKLML v0.14 z webu MKLML. Odsuďte ho do cesty MKLML a vytvořte v ní podadresáře verze. Sestavení MKL-DNN pomocí MKLML a instalace mkl-dnn do /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

Poznámka: Pokud chcete vytvořit s různými verzemi MKLML,

  • Při konfiguraci sestavení (cf. níže) zadejte možnost --with-mkl=<directory>, např --with-mkl=/usr/local/mkl/mklml_lnx_2018.0.3.20180406. .