From 64585dac7ef4a4c03a8a8700a8e75224fb4d0241 Mon Sep 17 00:00:00 2001 From: "Billy D." Date: Sun, 8 Feb 2026 21:39:11 -0500 Subject: [PATCH] fixing numpy pin. --- dockerfiles/Dockerfile.ray-worker-strixhalo | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/dockerfiles/Dockerfile.ray-worker-strixhalo b/dockerfiles/Dockerfile.ray-worker-strixhalo index eedb983..6098edb 100644 --- a/dockerfiles/Dockerfile.ray-worker-strixhalo +++ b/dockerfiles/Dockerfile.ray-worker-strixhalo @@ -80,9 +80,11 @@ RUN --mount=type=cache,target=/home/ray/.cache/uv,uid=1000,gid=1000 \ torch torchaudio torchvision # FIX: Ray base image has pandas 1.5.3 which is incompatible with numpy 2.x -# The TheRock PyTorch wheels require numpy 2.x, so upgrade pandas to match +# The TheRock PyTorch wheels require numpy 2.x, so upgrade pandas to match. +# Pin numpy <2.3 because numba (required by vLLM for speculative decoding) +# does not yet support numpy 2.3+. RUN --mount=type=cache,target=/home/ray/.cache/uv,uid=1000,gid=1000 \ - uv pip install --system 'pandas>=2.0.0,<3.0' + uv pip install --system 'pandas>=2.0.0,<3.0' 'numpy>=2.1.0,<2.3' # Install amdsmi sysfs shim LAST (required for vLLM ROCm platform detection). # The native amdsmi from ROCm 7.1 requires glibc 2.38 (Ubuntu 24.04),