a0d7b96a8f
1. Remove some useless installed packages 2. Adjust max number of threads according to CPU cores
41 lines
940 B
Bash
Executable File
41 lines
940 B
Bash
Executable File
#!/bin/bash
|
|
set -e
|
|
|
|
# Tell CMake to search for packages in Occlum toolchain's directory only
|
|
export PKG_CONFIG_LIBDIR=/usr/local/occlum/x86_64-linux-musl/lib
|
|
|
|
# Install the dependencies
|
|
apt-get update
|
|
apt-get install -y python3-pip python3-setuptools
|
|
pip3 install kubernetes
|
|
|
|
# Download and build XGBoost
|
|
rm -rf xgboost_src && mkdir xgboost_src
|
|
pushd xgboost_src
|
|
git clone https://github.com/dmlc/xgboost .
|
|
git checkout 6d5b34d82486cd1d0480c548f5d1953834659bd6
|
|
git submodule init
|
|
git submodule update
|
|
git apply ../patch/xgboost-01.diff
|
|
pushd rabit
|
|
git apply ../../patch/rabit-01.diff
|
|
popd
|
|
pushd dmlc-core
|
|
git apply ../../patch/dmlc-core-01.diff
|
|
popd
|
|
mkdir build
|
|
cd build
|
|
cmake ../ \
|
|
-DCMAKE_BUILD_TYPE=Release \
|
|
-DCMAKE_C_COMPILER=occlum-gcc -DCMAKE_CXX_COMPILER=occlum-g++
|
|
make -j4
|
|
popd
|
|
|
|
# Prepare data
|
|
pushd xgboost_src/demo/binary_classification
|
|
python mapfeat.py
|
|
python mknfold.py agaricus.txt 1
|
|
popd
|
|
|
|
echo "Build XGBoost Success!"
|