# 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.

if [ "$ANACONDA_PYTHON_VERSION" = "3.10" ]; then

conda_install numpy=1.21.2 astunparse mkl mkl-include setuptools cffi future six llvmdev=8.0.0 -c conda-forge

elif [ "$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 -c conda-forge

# Pin MyPy version because new errors are likely to appear with each release

# Pin hypothesis to avoid flakiness: https://github.com/pytorch/pytorch/issues/31136

conda_pip_install pytest \

scikit-image \

psutil \

unittest-xml-reporting \