Pytorch DDP on Databricks
Hello!I am currently trying to use Pytorch Lightning inside Databricks and I am currently using a cluster with 2 gpus. Whenever I try to train my Transformer model with 1 gpu in DP strategy everything works fine, but when I try to use both the 2 gpus...
- 1556 Views
- 1 replies
- 2 kudos
Latest Reply
Hi @Marco Capusso​ , I am facing the similar issue could you find some fix. It would be great if you share some details around it.
- 2 kudos