Merge pull request #245 from aws-samples/sean-smith-patch-2

Activate conda environment
aws-samples · Apr 5, 2024 · 81f2d50 · 81f2d50
2 parents 6bc1417 + bd172e0
commit 81f2d50
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/3.test_cases/13.SM-dataparallel-deepspeed/exec_torchrun.sh b/3.test_cases/13.SM-dataparallel-deepspeed/exec_torchrun.sh
@@ -3,6 +3,10 @@
 # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 # SPDX-License-Identifier: MIT-0
 
+# Activate conda environment
+source ./conda/bin/activate
+conda activate smdataparallel
+
 export OMP_NUM_THREADS=1
 export GPUS_PER_NODE=8 # p4d/p4de instances have 8 GPUs per node
 MASTER_NODE=$(scontrol show hostname | head -n 1)