Skip to content

Commit

Permalink
Fix MKL setup on arm
Browse files Browse the repository at this point in the history
  • Loading branch information
FranekStark committed Feb 11, 2025
1 parent 16008b9 commit 4cc713c
Show file tree
Hide file tree
Showing 2 changed files with 22 additions and 5 deletions.
8 changes: 3 additions & 5 deletions docker/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -79,11 +79,9 @@ ENV ACADOS_SOURCE_DIR "/root/acados"

# Install Intel MKL
WORKDIR /root
RUN apt-get update && apt-get install -y gpg-agent wget
RUN wget -O- https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRODUCTS.PUB | gpg --dearmor | sudo tee /usr/share/keyrings/oneapi-archive-keyring.gpg > /dev/null
RUN echo "deb [signed-by=/usr/share/keyrings/oneapi-archive-keyring.gpg] https://apt.repos.intel.com/oneapi all main" | sudo tee /etc/apt/sources.list.d/oneAPI.list
RUN apt-get update && apt-get install -y intel-oneapi-mkl
RUN echo 'source /opt/intel/oneapi/mkl/latest/env/vars.sh' >> ~/.bashrc
COPY mkl_setup.sh /mkl_setup.sh
RUN /mkl_setup.sh
RUN rm /mkl_setup.sh

# Install unitree_sdk2
RUN apt-get update && apt-get install net-tools
Expand Down
19 changes: 19 additions & 0 deletions docker/mkl_setup.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
#!/bin/sh

# perform a binary install on x86 machines
# on ARM, it is skipped

if [ "${HW_ARCH}" = "x86_64" ]; then
cd /root
apt-get update && apt-get install -y gpg-agent wget
wget -O- https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRODUCTS.PUB | gpg --dearmor | sudo tee /usr/share/keyrings/oneapi-archive-keyring.gpg > /dev/null
echo "deb [signed-by=/usr/share/keyrings/oneapi-archive-keyring.gpg] https://apt.repos.intel.com/oneapi all main" | sudo tee /etc/apt/sources.list.d/oneAPI.list
apt-get update && apt-get install -y intel-oneapi-mkl
echo 'source /opt/intel/oneapi/mkl/latest/env/vars.sh' >> ~/.bashrc

elif [ "${HW_ARCH}" = "aarch64" ]; then
echo 'Skipping MKL on ARM'
fi


exit 0

0 comments on commit 4cc713c

Please sign in to comment.