mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-05-01 19:17:41 +03:00
Provide better support for running embeddings on GPU
This commit is contained in:
parent
931fbc2c22
commit
512d5f6190
@ -63,6 +63,7 @@ COPY --from=rocm /opt/rocm-dist/ /
|
||||
FROM deps-prelim AS rocm-prelim-hsa-override0
|
||||
ENV HSA_ENABLE_SDMA=0
|
||||
ENV MIGRAPHX_ENABLE_NHWC=1
|
||||
ENV TF_ROCM_USE_IMMEDIATE_MODE=1
|
||||
|
||||
COPY --from=rocm-dist / /
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user