Skip to content

Instantly share code, notes, and snippets.

@thomasaarholt

thomasaarholt/xgboost_gpu.sh

Last active July 24, 2022 11:01
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save thomasaarholt/40a08720566d59a1a58dfecafdecea06 to your computer and use it in GitHub Desktop.
Save thomasaarholt/40a08720566d59a1a58dfecafdecea06 to your computer and use it in GitHub Desktop.
Download ZIP
Building XGBoost with GPU support from source on Red Hat with mload
Raw
xgboost_gpu.sh
# must not use gcc version > 10
module load CUDA/11.0 CMake/3.21.1-GCCcore-11.2.0
# Clone with recursive! Otherwise you get dmlc error later on!
git clone --recursive https://github.com/dmlc/xgboost
cd xgboost/python-package
# clean out any old build files that may break things if try to recompile
rm -rf build
# Build with CUDA support:
python setup.py install --use-cuda --use-nccl
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment