fix cuda number

0584ef40 · root · 39aef4f6 · 0584ef40
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

run.sh run.sh +1 -1

未找到文件。
--- a/run.sh
+++ b/run.sh
@@ -4,7 +4,7 @@
 export FLAGS_sync_nccl_allreduce=0
 export FLAGS_eager_delete_tensor_gb=1

-export CUDA_VISIBLE_DEVICES=2
+export CUDA_VISIBLE_DEVICES=0

 if [[ ! -d pretrain_model/bert ]]; then
    bash download_pretrain.sh bert