Update multi_node.sh

e5910c48 · zhougaofeng · 8d254593 · e5910c48
Commit e5910c48 authored Jul 01, 2024 by zhougaofeng
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

finetune/multi_node.sh finetune/multi_node.sh +1 -1

No files found.
--- a/finetune/multi_node.sh
+++ b/finetune/multi_node.sh
@@ -8,7 +8,7 @@ RANK=0
 MASTER_ADDR=127.0.0.1
 MASTER_PORT=17170

-CUDA_VISIBLE_DEVICES=6,7 torchrun \
+HIP_VISIBLE_DEVICES=6,7 torchrun \
    --nproc_per_node $NPROC_PER_NODE \
    --nnodes $NNODES \
    --node_rank $RANK \