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 \
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
||||||
python3 python3-pip ffmpeg \
|
python3 python3-pip ffmpeg \
|
||||||
|
hipblas miopen-hip rocrand \
|
||||||
&& rm -rf /var/lib/apt/lists/* \
|
&& rm -rf /var/lib/apt/lists/* \
|
||||||
&& pip install --no-cache-dir /wheels/*.whl \
|
&& pip install --no-cache-dir /wheels/*.whl \
|
||||||
&& pip install --no-cache-dir \
|
&& pip install --no-cache-dir \
|
||||||
|
|||||||
Reference in New Issue
Block a user