From 3319f6ca7891c53b267ae6a6beb65d128362e8f5 Mon Sep 17 00:00:00 2001 From: Genghong Hu <111997394+GodHu777777@users.noreply.github.com> Date: Wed, 16 Sep 2026 10:37:50 +0800 Subject: [PATCH] [CI/Build][Hardware][Ascend] Add torchmetrics to the NPU CI image (#7570) Signed-off-by: Genghong Hu --- docker/Dockerfile.npu.ci | 1 + 1 file changed, 1 insertion(+) diff --git a/docker/Dockerfile.npu.ci b/docker/Dockerfile.npu.ci index b65b803d6..a25155111 100644 --- a/docker/Dockerfile.npu.ci +++ b/docker/Dockerfile.npu.ci @@ -44,6 +44,7 @@ RUN python3 -m pip install -U pip && \ "onnx>=1.16.0" \ "librosa>=0.10.0" \ "datasets>=3.0.0" \ + "torchmetrics==1.9.0" \ --index-url http://cache-service.nginx-pypi-cache.svc.cluster.local/pypi/simple \ --trusted-host cache-service.nginx-pypi-cache.svc.cluster.local \ --default-timeout=60