XGBoost model: train on GPU, run on CPU without GPU RAM allocation

后端 未结 0 1826
爱一瞬间的悲伤
爱一瞬间的悲伤 2020-12-03 23:58

How can I train an XGBoost model on a GPU but run predictions on CPU without allocating any GPU RAM?

My situation: I create an XGBoot model

相关标签:
回答
  • 消灭零回复
提交回复
热议问题