Taylor pruner under-utilizing resources

[marianbasti]

Hi there, nice work!

I've been tinkering with the repo, and came across some issues when trying to fully utilize the available resources.

For example, I've learned that there is no way yet to distribute the pruning process over multiple GPUs (or haven't found a way yet), and the ones I have available can't fit all in 24gb, so i falloff to set the device to cpu .

Then I can kickstart the process no problem, but when I go check the resource usage, I see that only one core of the CPU -one thread- is actually doing all the work, leading to a extremely slow performance.

Is there a setting that i'm missing?

Cheers!