Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-10-27 10:52:11 +01:00
Code Issues Projects Releases Wiki Activity

docker/rocm: compile onnx files into mxr files at startup

Browse Source
This commit is contained in:
Indrek Mandre 2024-02-04 13:52:27 +02:00
parent b151bcca09
commit 8a14703d26
7 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
docker/rocm/Dockerfile
View File

@ -60,6 +60,7 @@ RUN apt-get -y install libnuma1
WORKDIR /opt/frigate/
COPY --from=rootfs / /
COPY docker/rocm/rootfs/ /
#######################################################################
FROM deps-prelim AS rocm-prelim-hsa-override0

0
docker/rocm/rootfs/etc/s6-overlay/s6-rc.d/compile-rocm-models/dependencies.d/download-models Normal file
View File

13
docker/rocm/rootfs/etc/s6-overlay/s6-rc.d/compile-rocm-models/run Executable file
View File

@ -0,0 +1,13 @@
#!/command/with-contenv bash
# shellcheck shell=bash
# Compile YoloV8 ONNX files into ROCm MIGraphX files
for onnx in /config/model_cache/yolov8/*.onnx
do
mxr="${onnx%.onnx}.mxr"
if ! test -f $mxr; then
echo "processing $onnx into $mxr"
/opt/rocm/bin/migraphx-driver compile $onnx --optimize --gpu --enable-offload-copy --binary -o $mxr
fi
done

1
docker/rocm/rootfs/etc/s6-overlay/s6-rc.d/compile-rocm-models/timeout-up Normal file
View File

@ -0,0 +1 @@
0

1
docker/rocm/rootfs/etc/s6-overlay/s6-rc.d/compile-rocm-models/type Normal file
View File

@ -0,0 +1 @@
oneshot

1
docker/rocm/rootfs/etc/s6-overlay/s6-rc.d/compile-rocm-models/up Normal file
View File

@ -0,0 +1 @@
/etc/s6-overlay/s6-rc.d/compile-rocm-models/run

0
docker/rocm/rootfs/etc/s6-overlay/s6-rc.d/frigate/dependencies.d/compile-rocm-models Normal file
View File