Skip to content

Commit

Permalink
Image for aarch64
Browse files Browse the repository at this point in the history
  • Loading branch information
TLCFEM committed Dec 11, 2024
1 parent 2e5aa44 commit 98adad4
Showing 1 changed file with 32 additions and 0 deletions.
32 changes: 32 additions & 0 deletions Script/Rocky.aarch64.Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,32 @@
FROM arm64v8/rockylinux:9 AS build

RUN dnf install -y epel-release && crb enable
RUN dnf install -y gcc g++ gfortran cmake wget git hdf5-devel

# part 1: openblas
RUN git clone --depth 1 --branch v0.3.28 https://github.com/OpenMathLib/OpenBLAS.git

# change the following configurations to match your needs
RUN cd OpenBLAS && make TARGET=ARMV8 DYNAMIC_ARCH=1 BINARY=64 USE_THREAD=1 USE_OPENMP=1 NUM_THREADS=20 NO_SHARED=1 NO_CBLAS=1 NO_LAPACKE=1 GEMM_MULTITHREAD_THRESHOLD=64

# part 2: tbb
RUN git clone --depth 1 --branch v2021.12.0 https://github.com/oneapi-src/oneTBB.git
RUN mkdir tbb-build && cd tbb-build && cmake -DCMAKE_BUILD_TYPE=Release -DTBB_TEST=OFF ../oneTBB && cmake --build . --target install --config Release --parallel "$(nproc)"

# part 3: suanpan
RUN git clone ---branch dev --depth 1 https://github.com/TLCFEM/suanPan.git

RUN cd suanPan && mkdir build && cd build && cmake -DCMAKE_BUILD_TYPE=Release -DUSE_SYS_LIB=ON -DCUSTOM_OPENBLAS=/OpenBLAS -DBUILD_MULTITHREAD=ON -DUSE_HDF5=ON -DUSE_VTK=OFF -DUSE_AVX2=OFF .. && make -j"$(nproc)"

FROM arm64v8/rockylinux:9 AS runtime

RUN dnf install -y epel-release && crb enable
RUN dnf install -y libgomp hdf5

COPY --from=build /tbb-build /tbb-build

RUN cd /tbb-build && make install && cd .. && rm -rf tbb-build

COPY --from=build /suanPan/build/suanPan /usr/local/bin/suanPan

RUN suanPan -v

0 comments on commit 98adad4

Please sign in to comment.