From 3adefb551b323b8076c7df7e2fa25db856e9b7a4 Mon Sep 17 00:00:00 2001 From: scott Date: Thu, 2 Apr 2026 22:04:52 -0400 Subject: [PATCH] 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 --- Dockerfile | 1 + 1 file changed, 1 insertion(+) diff --git a/Dockerfile b/Dockerfile index fc1ca26..cf8dfad 100644 --- a/Dockerfile +++ b/Dockerfile @@ -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 \