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 \