From bd3c144e0b8e82c9b3c5c40c6d557fe8665de5a3 Mon Sep 17 00:00:00 2001 From: Woosuk Kwon Date: Wed, 10 Apr 2024 07:37:17 -0700 Subject: [PATCH] [Bugfix][ROCm] Add numba to Dockerfile.rocm (#3962) --- Dockerfile.rocm | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile.rocm b/Dockerfile.rocm index 10b8bf1e7fabd..b1c5fac9d78ef 100644 --- a/Dockerfile.rocm +++ b/Dockerfile.rocm @@ -91,7 +91,7 @@ RUN if [ "$BUILD_TRITON" = "1" ]; then \ COPY ./ /app/vllm -RUN python3 -m pip install --upgrade pip +RUN python3 -m pip install --upgrade pip numba RUN python3 -m pip install xformers==0.0.23 --no-deps RUN cd /app \