We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 020f29b commit 634b8f6Copy full SHA for 634b8f6
src/sagemaker/remote_function/job.py
@@ -200,7 +200,8 @@
200
fi
201
202
printf "INFO: Invoking remote function with torchrun inside conda environment: $conda_env.\\n"
203
- $conda_exe run -n $conda_env torchrun --nproc_per_node $NPROC_PER_NODE -m sagemaker.remote_function.invoke_function "$@"
+ $conda_exe run -n $conda_env torchrun --nproc_per_node $NPROC_PER_NODE \
204
+ -m sagemaker.remote_function.invoke_function "$@"
205
else
206
printf "INFO: No conda env provided. Invoking remote function with torchrun\\n"
207
torchrun --nproc_per_node $NPROC_PER_NODE -m sagemaker.remote_function.invoke_function "$@"
0 commit comments