Skip to content

GitLab

"docs/git@developer.sourcefind.cn:OpenDAS/deepspeed.git" did not exist on "e351090c6cbf3f92f7f875ce4233230d936f2baa"
Unverified Commit bca7a494 authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

Remove pinning of mpmath (#8298)

parent fa5b8446
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
.github/scripts/setup-env.sh
View file @ bca7a494
...@@ -87,8 +87,7 @@ case $GPU_ARCH_TYPE in ...@@ -87,8 +87,7 @@ case $GPU_ARCH_TYPE in
;; ;;
esac esac
PYTORCH_WHEEL_INDEX="https://download.pytorch.org/whl/${CHANNEL}/${GPU_ARCH_ID}" 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 --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 if [[ $GPU_ARCH_TYPE == 'cuda' ]]; then
python -c "import torch; exit(not torch.cuda.is_available())" python -c "import torch; exit(not torch.cuda.is_available())"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment