File tree Expand file tree Collapse file tree 1 file changed +0
-4
lines changed
src/sagemaker/remote_function Expand file tree Collapse file tree 1 file changed +0
-4
lines changed Original file line number Diff line number Diff line change 159159
160160 printf "INFO: Invoking remote function inside conda environment: $conda_env.\\ n"
161161 printf "INFO: $conda_exe run -n $conda_env python -m sagemaker.remote_function.invoke_function \\ n"
162-
163162 $conda_exe run -n $conda_env python -m sagemaker.remote_function.invoke_function "$@"
164163else
165164 printf "INFO: No conda env provided. Invoking remote function\\ n"
166165 printf "INFO: python -m sagemaker.remote_function.invoke_function \\ n"
167-
168166 python -m sagemaker.remote_function.invoke_function "$@"
169167fi
170168"""
213211 printf "INFO: $conda_exe run -n $conda_env torchrun --nnodes $SM_HOST_COUNT --nproc_per_node $SM_NPROC_PER_NODE \
214212 --master_addr $SM_MASTER_ADDR --master_port $SM_MASTER_PORT --node_rank $SM_CURRENT_HOST_RANK \
215213 -m sagemaker.remote_function.invoke_function \\ n"
216-
217214 $conda_exe run -n $conda_env torchrun --nnodes $SM_HOST_COUNT --nproc_per_node $SM_NPROC_PER_NODE \
218215 --master_addr $SM_MASTER_ADDR --master_port $SM_MASTER_PORT --node_rank $SM_CURRENT_HOST_RANK \
219216 -m sagemaker.remote_function.invoke_function "$@"
220217else
221218 printf "INFO: No conda env provided. Invoking remote function with torchrun\\ n"
222219 printf "INFO: torchrun --nnodes $SM_HOST_COUNT --nproc_per_node $SM_NPROC_PER_NODE --master_addr $SM_MASTER_ADDR \
223220 --master_port $SM_MASTER_PORT --node_rank $SM_CURRENT_HOST_RANK -m sagemaker.remote_function.invoke_function \\ n"
224-
225221 torchrun --nnodes $SM_HOST_COUNT --nproc_per_node $SM_NPROC_PER_NODE --master_addr $SM_MASTER_ADDR \
226222 --master_port $SM_MASTER_PORT --node_rank $SM_CURRENT_HOST_RANK -m sagemaker.remote_function.invoke_function "$@"
227223fi
You can’t perform that action at this time.
0 commit comments