We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3b5e6fc commit c8c3839Copy full SHA for c8c3839
.github/scripts/setup-env.sh
@@ -87,7 +87,8 @@ case $GPU_ARCH_TYPE in
87
;;
88
esac
89
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}"
+# 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}"
92
93
if [[ $GPU_ARCH_TYPE == 'cuda' ]]; then
94
python -c "import torch; exit(not torch.cuda.is_available())"
0 commit comments