diff --git a/Dockerfile b/Dockerfile index d0a25dcca046..7bc6b200e87f 100644 --- a/Dockerfile +++ b/Dockerfile @@ -67,8 +67,8 @@ ARG CUDA_VERSION=11.7 ARG CUDA_CHANNEL=nvidia ARG INSTALL_CHANNEL=pytorch-nightly # Automatically set by buildx -RUN /opt/conda/bin/conda update -y conda -RUN /opt/conda/bin/conda install -c "${INSTALL_CHANNEL}" -y python=${PYTHON_VERSION} +# Note conda needs to be pinned to 23.5.2 see: https://github.com/pytorch/pytorch/issues/106470 +RUN /opt/conda/bin/conda install -c "${INSTALL_CHANNEL}" -y python=${PYTHON_VERSION} conda=23.5.2 ARG TARGETPLATFORM # On arm64 we can only install wheel packages.