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