Infrastructure/immich
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2025-03-18 02:31:28 -05:00
Code Issues Activity

set migraphx_home

Browse source
This commit is contained in:
mertalev 2025-03-05 20:49:58 -05:00
parent 521f9fb72d
commit c121d3e487
No known key found for this signature in database
GPG key ID: 3A2B5BFC678DBC80
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
machine-learning/Dockerfile
View file

@ -40,7 +40,7 @@ RUN git apply /tmp/0001-disable-rocm-conv-algo-caching.patch
RUN /bin/sh ./dockerfiles/scripts/install_common_deps.sh
# Note: the `parallel` setting uses a substantial amount of RAM
RUN ./build.sh --allow_running_as_root --config Release --build_wheel --update --build --parallel 17 --cmake_extra_defines\
ONNXRUNTIME_VERSION=1.20.1 --use_rocm --rocm_home=/opt/rocm --use_migraphx
ONNXRUNTIME_VERSION=1.20.1 --use_migraphx --migraphx_home /opt/rocm
RUN mv /code/onnxruntime/build/Linux/Release/dist/*.whl /opt/
FROM builder-${DEVICE} AS builder