diff --git a/common/install_mkl.sh b/common/install_mkl.sh index 5ebdd94b1..5889dc1f0 100644 --- a/common/install_mkl.sh +++ b/common/install_mkl.sh @@ -3,8 +3,8 @@ set -ex # MKL -MKL_VERSION=2020.0 -MKL_BUILD=166 +MKL_VERSION=2022.2.1 +MKL_BUILD=16993 mkdir -p /opt/intel/lib pushd /tmp curl -fsSL https://anaconda.org/intel/mkl-static/${MKL_VERSION}/download/linux-64/mkl-static-${MKL_VERSION}-intel_${MKL_BUILD}.tar.bz2 | tar xjv diff --git a/wheel/build_wheel.sh b/wheel/build_wheel.sh index 08b47335a..b8b663ac0 100755 --- a/wheel/build_wheel.sh +++ b/wheel/build_wheel.sh @@ -168,7 +168,7 @@ conda create ${EXTRA_CONDA_INSTALL_FLAGS} -yn "$tmp_env_name" python="$desired_p source activate "$tmp_env_name" 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 -retry conda install ${EXTRA_CONDA_INSTALL_FLAGS} -yq mkl-include==2020.1 mkl-static==2020.1 -c intel +retry conda install ${EXTRA_CONDA_INSTALL_FLAGS} -yq mkl-include==2022.2.1 mkl-static==2022.2.1 -c intel retry pip install -qr "${pytorch_rootdir}/requirements.txt" || true # For USE_DISTRIBUTED=1 on macOS, need libuv and pkg-config to find libuv.