yeyupiaoling/PP-YOLOE

多卡训练该如何配置

yihuiluo235 opened this issue · 3 comments

yihuiluo235 commented

export CUDA_VISIBLE_DEVICES=0,1 python3 -m paddle.distributed.launch --gpus 0,1 train.py --eval

执行上面的命令报错，麻烦看看，该如何调整支持多卡
image

yeyupiaoling commented

你有两张显卡吗？都正常吗？

yihuiluo235 commented

嗯，有两张，都正常的，也装了nccl2
image

yeyupiaoling commented

我也看不出是什么原因