cjones/immich
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

use composite cache key

Browse Source 
1.19.2

fix variable name

fix variable reference

aaaaaaaaaaaaaaaaaaaa
This commit is contained in:
mertalev
2024-12-30 17:02:35 -05:00
parent ec0eb93036
commit fe2ddc3644
3 changed files with 155 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
machine-learning/Dockerfile
View File

@@ -17,6 +17,7 @@ RUN mkdir /opt/armnn && \
# Warning: 26.3Gb of disk space required to pull this image
# https://github.com/microsoft/onnxruntime/blob/main/dockerfiles/Dockerfile.rocm
# 6.2 or later fails to build as of writing
FROM rocm/dev-ubuntu-22.04:6.1.2-complete AS builder-rocm
WORKDIR /code
@@ -32,11 +33,12 @@ RUN wget -nv https://github.com/Kitware/CMake/releases/download/v3.27.3/cmake-3.
ENV PATH /code/cmake-3.27.3-linux-x86_64/bin:${PATH}
# Prepare onnxruntime repository & build onnxruntime
# 1.20.1 fails to build as of writing
RUN git clone --single-branch --branch v1.19.2 --recursive "https://github.com/Microsoft/onnxruntime" onnxruntime
WORKDIR /code/onnxruntime
# Fix for multi-threading based on comments in https://github.com/microsoft/onnxruntime/pull/19567
COPY ./0001-guard-algo-benchmark-results.patch /tmp/
RUN git apply /tmp/0001-guard-algo-benchmark-results.patch
COPY ./0001-fix-rocm-conv-thread-safety.patch /tmp/
RUN git apply /tmp/0001-fix-rocm-conv-thread-safety.patch
RUN /bin/sh ./dockerfiles/scripts/install_common_deps.sh
# Note: the `parallel` setting uses a substantial amount of RAM
@@ -112,7 +114,7 @@ COPY --from=builder-armnn \
/opt/ann/build.sh \
/opt/armnn/
FROM rocm/dev-ubuntu-24.04:6.2.4-complete AS prod-rocm
FROM rocm/dev-ubuntu-22.04:6.1.2-complete AS prod-rocm
FROM prod-${DEVICE} AS prod