Skip Navigation
Machine Learning @lemmy.ml kernelPanic @lemmy.ml

[Solved] PyTorch Lightning is bottlenecked by the CPU

When I train my PyTorch Lightning model on two GPUs on jupyter lab with strategy="ddp_notebook", only two CPUs are used and their usages are 100%. How can I overcome this CPU bottleneck?

Edit: I tested with PyTorchProfiler and it was because of old ssds used on the server

2
2 comments