XMLWordPrintableJSON

Details

    • Sub-task
    • Status: Open
    • Major
    • Resolution: Unresolved
    • 3.4.0
    • None
    • ML, PySpark
    • None

    Description

      We want the distributor to have the ability to run multiple torchrun processes per task if task.gpu.amount > 1.

      We want to add a check to see if `import torch` doesn't raise an ImportError since the TorchDistributor requires torch. If it raises an ImportError, we will give the user more details. 

      Attachments

        Activity

          People

            Unassigned Unassigned
            erithwik Rithwik Ediga Lakhamsani
            Votes:
            0 Vote for this issue
            Watchers:
            1 Start watching this issue

            Dates

              Created:
              Updated: