Skip to content

Commit c8c3839

Browse files
authored
Pin 'mpmath<1.4' in CI (#8291)
1 parent 3b5e6fc commit c8c3839

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

.github/scripts/setup-env.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,8 @@ case $GPU_ARCH_TYPE in
8787
;;
8888
esac
8989
PYTORCH_WHEEL_INDEX="https://download.pytorch.org/whl/${CHANNEL}/${GPU_ARCH_ID}"
90-
pip install --progress-bar=off --pre torch --index-url="${PYTORCH_WHEEL_INDEX}"
90+
# TODO: remove pinning of mpmath when https://github.com/pytorch/vision/issues/8292 is properly fixed.
91+
pip install --progress-bar=off "mpmath<1.4" --pre torch --index-url="${PYTORCH_WHEEL_INDEX}"
9192

9293
if [[ $GPU_ARCH_TYPE == 'cuda' ]]; then
9394
python -c "import torch; exit(not torch.cuda.is_available())"

0 commit comments

Comments
 (0)