fix env

abcucberkeley · Oct 28, 2024 · df37b25 · df37b25
1 parent a311242
commit df37b25
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/ray_lsf_cluster.sh b/src/ray_lsf_cluster.sh
@@ -68,7 +68,7 @@ export head_node
 export head_node_ip
 export cluster_address
 
-bash ray_start_cluster.sh -e $env -i $head_node_ip -p $port -d $dashboard_port -c $cpus -g $gpus &
+bash ray_start_cluster.sh -e "${env}" -i $head_node_ip -p $port -d $dashboard_port -c $cpus -g $gpus &
 sleep 10
 
 ############################## ADD WORKER NODES
@@ -78,7 +78,7 @@ num_workers=$((nodes - 1))
 for i in $(seq 1 $num_workers)
 do
     echo "Adding worker: ${outdir}/ray_worker_${i}"
-    worker="bash ray_start_worker.sh -e $env -a $cluster_address -c $cpus -g $gpus"
+    worker="bash ray_start_worker.sh -e "${env}" -a $cluster_address -c $cpus -g $gpus"
     job="bsub -cwd "$(pwd)" -q $LSB_QUEUE -J "${outdir}/ray_worker_${i}" -n $cpus -gpu "num=$gpus:mode=shared" -o "${outdir}/ray_worker_${i}.log" $worker"
     echo $job
     $job
@@ -97,7 +97,7 @@ done
 
 ############################## CHECK STATUS
 
-bash ray_check_status.sh -e $env -a $cluster_address -r $nodes
+bash ray_check_status.sh -e "${env}" -a $cluster_address -r $nodes
 
 ############################## RUN WORKLOAD