restrict onnxruntime to amd64

ika-rwth-aachen · Dec 5, 2024 · b1df059 · b1df059
1 parent 8f0902c
commit b1df059
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -266,7 +266,9 @@ RUN if [[ -n $TF_VERSION ]]; then \
 # install ONNX Runtime
 ARG ONNX_RUNTIME_VERSION
 RUN if [[ -n $ONNX_RUNTIME_VERSION ]]; then \
-        pip3 install onnxruntime-gpu==${ONNX_RUNTIME_VERSION}; \
+        if [[ "$TARGETARCH" == "amd64" ]]; then \
+            pip3 install onnxruntime-gpu==${ONNX_RUNTIME_VERSION}; \
+        fi; \
     fi
 
 # --- finalization -------------------------------------------------------------------------------