Skip to content

Commit 11b96a7

Browse files
committed
Use install_conda.sh
1 parent 89c1603 commit 11b96a7

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

.ci/docker/common/install_conda.sh

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,8 +31,12 @@ install_miniconda() {
3131

3232
install_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

.ci/docker/conda-env-ci.txt

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,5 +3,3 @@ ninja=1.10.2
33
libuv
44
llvm-openmp
55
pkg-config
6-
mkl=2021.4.0
7-
mkl-include=2021.4.0

0 commit comments

Comments
 (0)