Install ROCm runtime libs (hipblas, miopen-hip, rocrand) in runtime stage

The builder had dev packages but the runtime stage was missing the .so
files, causing ImportError on libhipblas.so.2 at startup.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
2026-04-02 22:04:52 -04:00
parent c3e0e720ba
commit 3adefb551b

View File

@@ -91,6 +91,7 @@ COPY --from=builder /wheels /wheels
RUN apt-get update && apt-get install -y --no-install-recommends \
python3 python3-pip ffmpeg \
hipblas miopen-hip rocrand \
&& rm -rf /var/lib/apt/lists/* \
&& pip install --no-cache-dir /wheels/*.whl \
&& pip install --no-cache-dir \