From 566f10a9293425483ac47892e2a6cd4960177a59 Mon Sep 17 00:00:00 2001 From: "Chendi.Xue" Date: Wed, 9 Apr 2025 12:52:26 -0500 Subject: [PATCH] [CI]Fix hpu docker and numpy version for CI (#16355) Signed-off-by: Chendi Xue --- docker/Dockerfile.hpu | 2 +- requirements/hpu.txt | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/docker/Dockerfile.hpu b/docker/Dockerfile.hpu index 48211c88..224f142b 100644 --- a/docker/Dockerfile.hpu +++ b/docker/Dockerfile.hpu @@ -1,4 +1,4 @@ -FROM vault.habana.ai/gaudi-docker/1.19.1/ubuntu22.04/habanalabs/pytorch-installer-2.5.1:latest +FROM vault.habana.ai/gaudi-docker/1.20.1/ubuntu22.04/habanalabs/pytorch-installer-2.6.0:latest COPY ./ /workspace/vllm diff --git a/requirements/hpu.txt b/requirements/hpu.txt index a61d72d0..830f6ef3 100644 --- a/requirements/hpu.txt +++ b/requirements/hpu.txt @@ -5,6 +5,7 @@ ray triton==3.1.0 pandas +numpy==1.26.4 tabulate setuptools>=61 setuptools-scm>=8