diff --git a/.circleci/docker/common/install_conda.sh b/.circleci/docker/common/install_conda.sh index cf8c108dd1a..1a5b79e2f48 100755 --- a/.circleci/docker/common/install_conda.sh +++ b/.circleci/docker/common/install_conda.sh @@ -73,19 +73,18 @@ if [ -n "$ANACONDA_PYTHON_VERSION" ]; then } # Install PyTorch conda deps, as per https://github.com/pytorch/pytorch README - # DO NOT install cmake here as it would install a version newer than 3.10, but - # we want to pin to version 3.10. + # DO NOT install cmake here as it would install a version newer than 3.13, but + # we want to pin to version 3.13. + CONDA_COMMON_DEPS="astunparse pyyaml mkl=2022.0.1 mkl-include=2022.0.1 setuptools cffi future six" if [ "$ANACONDA_PYTHON_VERSION" = "3.9" ]; then # Install llvm-8 as it is required to compile llvmlite-0.30.0 from source - conda_install numpy=1.19.2 astunparse pyyaml mkl mkl-include setuptools cffi future six llvmdev=8.0.0 + conda_install numpy=1.19.2 ${CONDA_COMMON_DEPS} llvmdev=8.0.0 elif [ "$ANACONDA_PYTHON_VERSION" = "3.8" ]; then # Install llvm-8 as it is required to compile llvmlite-0.30.0 from source - conda_install numpy=1.18.5 astunparse pyyaml mkl mkl-include setuptools cffi future six llvmdev=8.0.0 - elif [ "$ANACONDA_PYTHON_VERSION" = "3.7" ]; then - # DO NOT install dataclasses if installing python-3.7, since its part of python-3.7 core packages - conda_install numpy=1.18.5 astunparse pyyaml mkl mkl-include setuptools cffi future six typing_extensions + conda_install numpy=1.18.5 ${CONDA_COMMON_DEPS} llvmdev=8.0.0 else - conda_install numpy=1.18.5 astunparse pyyaml mkl mkl-include setuptools cffi future six dataclasses typing_extensions + # Install `typing_extensions` for 3.7 + conda_install numpy=1.18.5 ${CONDA_COMMON_DEPS} typing_extensions fi # Magma package names are concatenation of CUDA major and minor ignoring revision