[ https://issues.apache.org/jira/browse/SPARK-41916?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel ]
Rithwik Ediga Lakhamsani updated SPARK-41916: --------------------------------------------- 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. was:We want the distributor to have the ability to run multiple torchrun processes per task if task.gpu.amount > 1. > Address General Fixes > --------------------- > > Key: SPARK-41916 > URL: https://issues.apache.org/jira/browse/SPARK-41916 > Project: Spark > Issue Type: Sub-task > Components: ML, PySpark > Affects Versions: 3.4.0 > Reporter: Rithwik Ediga Lakhamsani > Priority: Major > > 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. -- This message was sent by Atlassian Jira (v8.20.10#820010) --------------------------------------------------------------------- To unsubscribe, e-mail: issues-unsubscr...@spark.apache.org For additional commands, e-mail: issues-h...@spark.apache.org