Skip to content

Commit

Permalink
Update al2 dockerfile for knn build
Browse files Browse the repository at this point in the history
Signed-off-by: Junqiu Lei <junqiu@amazon.com>
  • Loading branch information
junqiu-lei committed May 23, 2024
1 parent c0d1aac commit 45f62e5
Showing 1 changed file with 12 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -90,12 +90,21 @@ RUN ln -sfn /usr/local/bin/python3.9 /usr/bin/python3 && \
pip3 install pip==23.1.2 && pip3 install pipenv==2023.6.12 awscli==1.32.17

# Add k-NN Library dependencies
RUN yum repolist && yum install openblas-static lapack gcc-gfortran -y
RUN yum repolist && yum install lapack gcc-gfortran -y && \
git clone -b v0.3.3 --single-branch https://github.com/xianyi/OpenBLAS.git && \
cd OpenBLAS && \
make USE_OPENMP=1 FC=gfortran && \
make PREFIX=/usr/local install
ENV LD_LIBRARY_PATH="/usr/local/lib:$LD_LIBRARY_PATH"

RUN pip3 install cmake==3.23.3
# Upgrade gcc
RUN yum install -y gcc10* && \
ln -s `which gcc10-gcc` /usr/local/bin/gcc && \
ln -s `which gcc10-g++` /usr/local/bin/g++
mv -v /usr/bin/gcc /usr/bin/gcc7-gcc && \
mv -v /usr/bin/g++ /usr/bin/gcc7-g++ && \
update-alternatives --install /usr/bin/gcc gcc $(which gcc10-gcc) 1 && \
update-alternatives --install /usr/bin/g++ g++ $(which gcc10-g++) 1
ENV FC=gcc10-gfortran

# Change User
USER $CONTAINER_USER
Expand Down

0 comments on commit 45f62e5

Please sign in to comment.