From b45457511a915c60c3f1d14060032d3a0d982c8f 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 | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/Dockerfile.rocm b/Dockerfile.rocm index 12cbea2d33091..464613dfb098d 100644 --- a/Dockerfile.rocm +++ b/Dockerfile.rocm @@ -91,8 +91,13 @@ RUN if [ "$BUILD_TRITON" = "1" ]; then \ COPY ./ /app/vllm +<<<<<<< HEAD RUN python3 -m pip install --upgrade pip RUN python3 -m pip install xformers==0.0.25 --no-deps +======= +RUN python3 -m pip install --upgrade pip numba +RUN python3 -m pip install xformers==0.0.23 --no-deps +>>>>>>> bd3c144e ([Bugfix][ROCm] Add numba to Dockerfile.rocm (#3962)) RUN cd /app \ && cd vllm \