From fb42b2fe780e16687f887486954776b8165d91e8 Mon Sep 17 00:00:00 2001
From: LDOUBLEV <liuvv0203@outlook.com>
Date: Thu, 15 Jul 2021 03:01:17 +0000
Subject: [PATCH] unset CUDA_VISIBLE_DEVICES before train

---
 tests/test.sh | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/test.sh b/tests/test.sh
index 60b0233c..8e8d012a 100644
--- a/tests/test.sh
+++ b/tests/test.sh
@@ -248,6 +248,7 @@ for gpu in ${gpu_list[*]}; do
                 cmd="${python} -m paddle.distributed.launch --ips=${ips} --gpus=${gpu} ${run_train} ${save_model_key}=${save_log} ${set_pretrain} ${set_epoch} ${set_autocast} ${set_batchsize} ${set_train_params1}"
             fi
             # run train
+            eval "unset CUDA_VISIBLE_DEVICES"
             eval $cmd
             status_check $? "${cmd}" "${status_log}"
 
-- 
GitLab