[Bugfix][ROCm] Add numba to Dockerfile.rocm (#3962)

2024-04-10 07:37:17 -07:00
parent 0258b7a94b
commit bd3c144e0b
1 changed files with 1 additions and 1 deletions
--- 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 \