diff --git a/docker/Dockerfile.rocm b/docker/Dockerfile.rocm index e60cf5e69a4c..7ba6236b1145 100644 --- a/docker/Dockerfile.rocm +++ b/docker/Dockerfile.rocm @@ -39,7 +39,7 @@ FROM fetch_vllm AS build_vllm ARG USE_CYTHON # Build vLLM RUN cd vllm \ - && python3 -m pip install -r requirements/rocm.txt \ + && python3 -m pip install -r requirements/rocm-build.txt -r requirements/rocm.txt \ && python3 setup.py clean --all \ && if [ ${USE_CYTHON} -eq "1" ]; then python3 tests/build_cython.py build_ext --inplace; fi \ && python3 setup.py bdist_wheel --dist-dir=dist diff --git a/requirements/rocm-build.txt b/requirements/rocm-build.txt index 981b90632c18..e422edd4ac8a 100644 --- a/requirements/rocm-build.txt +++ b/requirements/rocm-build.txt @@ -14,3 +14,4 @@ setuptools-scm>=8 wheel jinja2>=3.1.6 amdsmi==6.2.4 +regex