File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change 3
3
set -ex
4
4
5
5
# MKL
6
- MKL_VERSION=2020.0
7
- MKL_BUILD=166
6
+ MKL_VERSION=2021.1
7
+ MKL_BUILD=16993
8
8
mkdir -p /opt/intel/lib
9
9
pushd /tmp
10
10
curl -fsSL https://anaconda.org/intel/mkl-static/${MKL_VERSION} /download/linux-64/mkl-static-${MKL_VERSION} -intel_${MKL_BUILD} .tar.bz2 | tar xjv
Original file line number Diff line number Diff line change @@ -168,7 +168,7 @@ conda create ${EXTRA_CONDA_INSTALL_FLAGS} -yn "$tmp_env_name" python="$desired_p
168
168
source activate " $tmp_env_name "
169
169
170
170
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
172
172
retry pip install -qr " ${pytorch_rootdir} /requirements.txt" || true
173
173
174
174
# For USE_DISTRIBUTED=1 on macOS, need libuv and pkg-config to find libuv.
You can’t perform that action at this time.
0 commit comments