diff --git a/.riahub/workflows/train.yaml b/.riahub/workflows/train.yaml index fa07433..05b8697 100644 --- a/.riahub/workflows/train.yaml +++ b/.riahub/workflows/train.yaml @@ -361,7 +361,9 @@ jobs: # FAST-PATH: install CPU torch from pytorch.org/whl/cpu FIRST (~200MB). # This makes torch==X already-satisfied so requirements.txt does not # pull the 755MB manylinux wheel with bundled CUDA from PyPI default. - $PIP install --index-url https://download.pytorch.org/whl/cpu --extra-index-url https://pypi.org/simple torch==2.2.2 torchvision + $PIP install --index-url https://download.pytorch.org/whl/cpu --extra-index-url https://pypi.org/simple "numpy<2" "torch==2.2.2" torchvision + # Also ensure numpy<2 is pinned for the requirements.txt install below + $PIP install --upgrade --force-reinstall "numpy<2" INSTALLED_SOMETHING=0 if [[ -f requirements.txt ]]; then $PIP install -r requirements.txt