mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-11-30 02:48:45 +08:00
97 lines
4.7 KiB
Bash
97 lines
4.7 KiB
Bash
#!/usr/bin/env bash
|
|
|
|
# Licensed to the LF AI & Data foundation under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
function install_linux_deps() {
|
|
if [[ -x "$(command -v apt)" ]]; then
|
|
# for Ubuntu 20.04
|
|
sudo apt install -y g++ gcc make ccache libssl-dev zlib1g-dev libboost-regex-dev \
|
|
libboost-program-options-dev libboost-system-dev libboost-filesystem-dev \
|
|
libboost-serialization-dev python3-dev libboost-python-dev libcurl4-openssl-dev gfortran libtbb-dev libzstd-dev
|
|
# install OpenBLAS, this could take a while.
|
|
wget https://github.com/xianyi/OpenBLAS/archive/v0.3.9.tar.gz && \
|
|
tar zxvf v0.3.9.tar.gz && cd OpenBLAS-0.3.9 && \
|
|
sudo make TARGET=CORE2 DYNAMIC_ARCH=1 DYNAMIC_OLDER=1 USE_THREAD=0 USE_OPENMP=0 FC=gfortran CC=gcc COMMON_OPT="-O3 -g -fPIC" FCOMMON_OPT="-O3 -g -fPIC -frecursive" NMAX="NUM_THREADS=128" LIBPREFIX="libopenblas" INTERFACE64=0 NO_STATIC=1 && \
|
|
sudo make PREFIX=/usr NO_STATIC=1 install && \
|
|
cd .. && rm -rf OpenBLAS-0.3.9 && rm v0.3.9.tar.gz
|
|
elif [[ -x "$(command -v yum)" ]]; then
|
|
# for CentOS 7
|
|
sudo yum install -y epel-release centos-release-scl-rh && \
|
|
sudo yum install -y git make automake ccache openssl-devel zlib-devel libzstd-devel \
|
|
libcurl-devel python3-devel \
|
|
devtoolset-11-gcc devtoolset-11-gcc-c++ devtoolset-11-gcc-gfortran \
|
|
llvm-toolset-11.0-clang llvm-toolset-11.0-clang-tools-extra
|
|
|
|
echo "source scl_source enable devtoolset-11" | sudo tee -a /etc/profile.d/devtoolset-11.sh
|
|
echo "source scl_source enable llvm-toolset-11.0" | sudo tee -a /etc/profile.d/llvm-toolset-11.sh
|
|
echo "export CLANG_TOOLS_PATH=/opt/rh/llvm-toolset-11.0/root/usr/bin" | sudo tee -a /etc/profile.d/llvm-toolset-11.sh
|
|
source "/etc/profile.d/llvm-toolset-11.sh"
|
|
|
|
# install OpenBLAS, this could take a while.
|
|
source "/etc/profile.d/devtoolset-11.sh" && \
|
|
wget https://github.com/xianyi/OpenBLAS/archive/v0.3.9.tar.gz && \
|
|
tar zxvf v0.3.9.tar.gz && cd OpenBLAS-0.3.9 && \
|
|
make TARGET=CORE2 DYNAMIC_ARCH=1 DYNAMIC_OLDER=1 USE_THREAD=0 USE_OPENMP=0 FC=gfortran CC=gcc COMMON_OPT="-O3 -g -fPIC" FCOMMON_OPT="-O3 -g -fPIC -frecursive" NMAX="NUM_THREADS=128" LIBPREFIX="libopenblas" INTERFACE64=0 NO_STATIC=1 && \
|
|
make PREFIX=/usr NO_STATIC=1 install && \
|
|
cd .. && rm -rf OpenBLAS-0.3.9 && rm v0.3.9.tar.gz
|
|
|
|
# Install tbb
|
|
git clone https://github.com/wjakob/tbb.git && \
|
|
cd tbb/build && \
|
|
cmake .. && make -j && \
|
|
sudo make install && \
|
|
cd ../../ && rm -rf tbb/
|
|
|
|
# Install boost
|
|
wget -q https://boostorg.jfrog.io/artifactory/main/release/1.65.1/source/boost_1_65_1.tar.gz && \
|
|
tar zxf boost_1_65_1.tar.gz && cd boost_1_65_1 && \
|
|
./bootstrap.sh --prefix=/usr/local --with-toolset=gcc --without-libraries=python && \
|
|
sudo ./b2 -j2 --prefix=/usr/local --without-python toolset=gcc install && \
|
|
cd ../ && rm -rf ./boost_1_65_1*
|
|
else
|
|
echo "Error Install Dependencies ..."
|
|
exit 1
|
|
fi
|
|
}
|
|
|
|
function install_mac_deps() {
|
|
sudo xcode-select --install > /dev/null 2>&1
|
|
brew install boost libomp ninja tbb cmake llvm ccache zstd
|
|
brew uninstall grep
|
|
brew install grep
|
|
export PATH="/usr/local/opt/grep/libexec/gnubin:$PATH"
|
|
brew update && brew upgrade && brew cleanup
|
|
|
|
if [[ $(arch) == 'arm64' ]]; then
|
|
sudo mkdir /usr/local/include
|
|
sudo mkdir /usr/local/opt
|
|
sudo ln -s "$(brew --prefix llvm)" "/usr/local/opt/llvm"
|
|
sudo ln -s "$(brew --prefix libomp)/include/omp.h" "/usr/local/include/omp.h"
|
|
sudo ln -s "$(brew --prefix libomp)" "/usr/local/opt/libomp"
|
|
sudo ln -s "$(brew --prefix boost)/include/boost" "/usr/local/include/boost"
|
|
sudo ln -s "$(brew --prefix tbb)/include/tbb" "/usr/local/include/tbb"
|
|
sudo ln -s "$(brew --prefix tbb)/include/oneapi" "/usr/local/include/oneapi"
|
|
fi
|
|
}
|
|
|
|
unameOut="$(uname -s)"
|
|
case "${unameOut}" in
|
|
Linux*) install_linux_deps;;
|
|
Darwin*) install_mac_deps;;
|
|
*) echo "Unsupported OS:${unameOut}" ; exit 0;
|
|
esac
|