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:
@@ -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 \
|
||||
|
||||
Reference in New Issue
Block a user