Distributed get_preds CUDA error

The solution was to follow this post instructions, mainly:

  • Delete dls and learner
  • Empty cuda cache
  • Call garbage collector