File tree Expand file tree Collapse file tree 2 files changed +6
-4
lines changed Expand file tree Collapse file tree 2 files changed +6
-4
lines changed Original file line number Diff line number Diff line change @@ -31,8 +31,12 @@ install_miniconda() {
3131
3232install_python () {
3333 pushd /opt/conda
34- # Install the correct Python version
35- as_ci_user conda create -n " py_${PYTHON_VERSION} " -y --file /opt/conda/conda-env-ci.txt python=" ${PYTHON_VERSION} "
34+ # Install the select Python version for CI jobs. Conda mkl is needed to build PyTorch
35+ # from source for optimal performance on x86 CPU
36+ as_ci_user conda create -n " py_${PYTHON_VERSION} " -y --file /opt/conda/conda-env-ci.txt \
37+ python=" ${PYTHON_VERSION} " \
38+ mkl=2021.4.0 \
39+ mkl-include=2021.4.0
3640 popd
3741}
3842
Original file line number Diff line number Diff line change @@ -3,5 +3,3 @@ ninja=1.10.2
33libuv
44llvm-openmp
55pkg-config
6- mkl=2021.4.0
7- mkl-include=2021.4.0
You can’t perform that action at this time.
0 commit comments