Skip to content

Commit a11530c

Browse files
committed
Update MKL to 2022.1
As previous one occasionally crashes on AMD CPUs
1 parent 3a407b2 commit a11530c

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

common/install_mkl.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@
33
set -ex
44

55
# MKL
6-
MKL_VERSION=2020.0
7-
MKL_BUILD=166
6+
MKL_VERSION=2021.1
7+
MKL_BUILD=16993
88
mkdir -p /opt/intel/lib
99
pushd /tmp
1010
curl -fsSL https://anaconda.org/intel/mkl-static/${MKL_VERSION}/download/linux-64/mkl-static-${MKL_VERSION}-intel_${MKL_BUILD}.tar.bz2 | tar xjv

wheel/build_wheel.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -168,7 +168,7 @@ conda create ${EXTRA_CONDA_INSTALL_FLAGS} -yn "$tmp_env_name" python="$desired_p
168168
source activate "$tmp_env_name"
169169

170170
retry conda install ${EXTRA_CONDA_INSTALL_FLAGS} -yq cmake "numpy${NUMPY_PINNED_VERSION}" nomkl "setuptools${SETUPTOOLS_PINNED_VERSION}" "pyyaml${PYYAML_PINNED_VERSION}" cffi typing_extensions ninja requests
171-
retry conda install ${EXTRA_CONDA_INSTALL_FLAGS} -yq mkl-include==2020.1 mkl-static==2020.1 -c intel
171+
retry conda install ${EXTRA_CONDA_INSTALL_FLAGS} -yq mkl-include==2022.1 mkl-static==2022.1 -c intel
172172
retry pip install -qr "${pytorch_rootdir}/requirements.txt" || true
173173

174174
# For USE_DISTRIBUTED=1 on macOS, need libuv and pkg-config to find libuv.

0 commit comments

Comments
 (0)