XGBoost using GPU

 

GPU로 XGBoost를 돌리려고 하다가 이것저것 문제가 발생했습니다.
해결하진 못했지만 과정을 기록해두었습니다.

자세한 설명은 official site를 참고

// CPU로 동작은 하나 GPU 사용에서 문제 발생
$ pip install xgboost


// 1. xgboost library 다운
$ git clone --recursive https://github.com/dmlc/xgboost.git

// 2. cmake 3.13.4 설치 (작동하는 version을 선택)
$ sudo apt remove cmake
$ wget https://cmake.org/files/v3.13/cmake-3.13.4.tar.gz
$ tar -zxvf cmake-3.13.4-Linux-x86_64.tar.gz
$ cd cmake-3.13.4-Linux-x86_64
$ ./bootstrap
$ make
$ sudo make install
$ source ~/.bashrc
$ cmake --version

// 3. nvcc (cuda-toolkit) 설치
$ sudo apt install nvidia-cuda-toolkit

// 4. gcc-5 설치
$ sudo apt install gcc-5
$ sudo apt install g++-5
$ sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 10
$ gcc --version
$ g++ --version

// 5. 드디어 설치 시작
$ cd xgboost
$ mkdir build
$ cd build
$ cmake .. -DUSE_CUDA=ON

// 실패..
$ make -j4

https://snowdeer.github.io/linux/2018/04/10/upgrade-cmake/ https://askubuntu.com/a/1087368