From 73509a9dcfecc9053a505dd2064014cffb79ac96 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Mon, 4 Mar 2024 13:49:54 +0000 Subject: [PATCH 1/2] Pin 'mpmath<1.4' in CI --- .github/scripts/setup-env.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/scripts/setup-env.sh b/.github/scripts/setup-env.sh index a4f113c367f..12d05ef7cba 100755 --- a/.github/scripts/setup-env.sh +++ b/.github/scripts/setup-env.sh @@ -87,7 +87,7 @@ case $GPU_ARCH_TYPE in ;; esac PYTORCH_WHEEL_INDEX="https://download.pytorch.org/whl/${CHANNEL}/${GPU_ARCH_ID}" -pip install --progress-bar=off --pre torch --index-url="${PYTORCH_WHEEL_INDEX}" +pip install --progress-bar=off "mpmath<1.4" --pre torch --index-url="${PYTORCH_WHEEL_INDEX}" if [[ $GPU_ARCH_TYPE == 'cuda' ]]; then python -c "import torch; exit(not torch.cuda.is_available())" From 785779a175507b3762f20ad00c104e0229356048 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Mon, 4 Mar 2024 13:52:24 +0000 Subject: [PATCH 2/2] Add comments --- .github/scripts/setup-env.sh | 1 + 1 file changed, 1 insertion(+) diff --git a/.github/scripts/setup-env.sh b/.github/scripts/setup-env.sh index 12d05ef7cba..d67c221b766 100755 --- a/.github/scripts/setup-env.sh +++ b/.github/scripts/setup-env.sh @@ -87,6 +87,7 @@ case $GPU_ARCH_TYPE in ;; esac PYTORCH_WHEEL_INDEX="https://download.pytorch.org/whl/${CHANNEL}/${GPU_ARCH_ID}" +# TODO: remove pinning of mpmath when https://github.com/pytorch/vision/issues/8292 is properly fixed. pip install --progress-bar=off "mpmath<1.4" --pre torch --index-url="${PYTORCH_WHEEL_INDEX}" if [[ $GPU_ARCH_TYPE == 'cuda' ]]; then