Xgboost doesn't release gpu memory after training/predicting the model on large data. Every further rerun of .fit causes more memory allocation until eventual crash of the kernel because GPU memory is out of bounds. Environment info Operating System: Ubuntu 16.04 on PowerPC Compiler: Package used (python/R/jvm/C++): python xgboost version used: If installing from source, please provide The commit