diff --git a/.riahub/workflows/train.yaml b/.riahub/workflows/train.yaml index 05b8697..a3c1537 100644 --- a/.riahub/workflows/train.yaml +++ b/.riahub/workflows/train.yaml @@ -380,6 +380,10 @@ jobs: exit 1 fi $PIP install h5py scipy + # After requirements.txt, force numpy back to <2 (torch 2.2.2 has + # NumPy 1.x ABI; transitive deps in requirements.txt would + # otherwise leave numpy 2.x in place and crash at runtime). + $PIP install --upgrade --force-reinstall "numpy<2" TORCH_INDEX_URL="https://download.pytorch.org/whl/cpu" TORCH_REASON="no NVIDIA GPU detected" if command -v nvidia-smi &> /dev/null; then