administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-05-03 12:07:40 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Provide better support for running embeddings on GPU

Browse Source
This commit is contained in:
Nicolas Mowen 2025-06-02 08:36:36 -06:00
parent 931fbc2c22
commit 512d5f6190
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
docker/rocm/Dockerfile
View File

@ -63,6 +63,7 @@ COPY --from=rocm /opt/rocm-dist/ /
FROM deps-prelim AS rocm-prelim-hsa-override0 FROM deps-prelim AS rocm-prelim-hsa-override0
ENV HSA_ENABLE_SDMA=0 ENV HSA_ENABLE_SDMA=0
ENV MIGRAPHX_ENABLE_NHWC=1 ENV MIGRAPHX_ENABLE_NHWC=1
ENV TF_ROCM_USE_IMMEDIATE_MODE=1
COPY --from=rocm-dist / / COPY --from=rocm-dist / /