Linux에서 MKL 설정

  • 아티클

기본 CNTK 수학 라이브러리는 Intel MKL(Intel Math Kernel Library)입니다. CNTK 이 리포지토리의 MKL-DNN뿐만 아니라 사용자 지정 라이브러리 버전 MKLML을 통해 Intel MKL 사용을 지원합니다.

MKL-DNN 및 MKLML 라이브러리 설치:

  • 머신에 MKLML을 저장할 디렉터리를 만듭니다. 예를 들면 다음과 같습니다.
sudo mkdir /usr/local/mklml
  • MKLML 웹 사이트에서 필요한 MKLML v0.14를 다운로드합니다. MKLML 경로에 타르를 빼고 버전이 지정된 하위 디렉터리를 만듭니다. MKLML을 사용하여 MKL-DNN 빌드 및 /usr/local에 mkl-dnn 설치
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).