From b92bc91beaca969f41af35768528dbfc9c341d13 Mon Sep 17 00:00:00 2001 From: Roman Pope Date: Thu, 28 May 2026 02:52:21 -0400 Subject: [PATCH] Workflow: pin numpy<2 (torch 2.2.2 ABI incompatible with numpy 2.x) --- .riahub/workflows/train.yaml | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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