Use uname -m for architecture

brian-team · Mar 11, 2024 · 84b2ad8 · 84b2ad8
1 parent f437aa6
commit 84b2ad8
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -26,15 +26,13 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 COPY dist /tmp/dist
 
 # Install Brian2 and recommended packages
-ARG TARGETARCH
 RUN python -m pip install --no-cache-dir --only-binary=:all: \
     scipy \
     matplotlib \
     jupyterlab \
     pytest \
     pytest-xdist \
-    && if [ "${TARGETARCH}" = "arm64" ]; then WHEEL_ARCH=aarch64; else WHEEL_ARCH=x86_64; fi \
-    && python -m pip install /tmp/dist/Brian2*_${WHEEL_ARCH}.whl brian2tools \
+    && python -m pip install /tmp/dist/Brian2*_$(uname -m).whl brian2tools \
     && rm -rf /tmp/dist
 
 # Create a non-root user