Fix

Signed-off-by: Joaquin Anton <[email protected]>
NVIDIA · Apr 19, 2024 · ae7dbcb · ae7dbcb
1 parent 27f3390
commit ae7dbcb
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/docker/Dockerfile.build.aarch64-linux b/docker/Dockerfile.build.aarch64-linux
@@ -59,12 +59,12 @@ RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/
     PYTHON_V=$(python3 -c "import sys;print(f'{sys.version_info[0]}.{sys.version_info[1]}')") && \
     ln -s /usr/bin/python${PYTHON_V}-config /usr/bin/python3-config
 
-RUN CMAKE_VERSION=3.20.1 && \
-    wget https://github.com/Kitware/CMake/releases/download/v${CMAKE_VERSION}/cmake-${CMAKE_VERSION}-linux-aarch64.sh && \
+RUN CMAKE_VERSION=3.20.1 && CMAKE_ARCH=$(uname -m) && \
+    wget https://github.com/Kitware/CMake/releases/download/v${CMAKE_VERSION}/cmake-${CMAKE_VERSION}-linux-${CMAKE_ARCH}.sh && \
     test -e /bin/sh || ln -s /usr/bin/sh /bin/sh && \
-    chmod +x cmake-${CMAKE_VERSION}-linux-aarch64.sh && \
-    ./cmake-${CMAKE_VERSION}-linux-aarch64.sh --prefix=/usr/local --skip-license && \
-    rm -rf cmake-${CMAKE_VERSION}-linux-aarch64.sh
+    chmod +x cmake-${CMAKE_VERSION}-linux-${CMAKE_ARCH}.sh && \
+    ./cmake-${CMAKE_VERSION}-linux-${CMAKE_ARCH}.sh --prefix=/usr/local --skip-license && \
+    rm -rf cmake-${CMAKE_VERSION}-linux-${CMAKE_ARCH}.sh
 
 ENV PKG_CONFIG_PATH=/usr/aarch64-linux-gnu/lib/pkgconfig