Skip to content

Commit bca7a49

Browse files
authored
Remove pinning of mpmath (#8298)
1 parent fa5b844 commit bca7a49

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

.github/scripts/setup-env.sh

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -87,8 +87,7 @@ case $GPU_ARCH_TYPE in
8787
;;
8888
esac
8989
PYTORCH_WHEEL_INDEX="https://download.pytorch.org/whl/${CHANNEL}/${GPU_ARCH_ID}"
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}"
90+
pip install --progress-bar=off --pre torch --index-url="${PYTORCH_WHEEL_INDEX}"
9291

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

0 commit comments

Comments
 (0)