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

vllm-project · Apr 10, 2024 · bd3c144 · bd3c144
1 parent 0258b7a
commit bd3c144
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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 \