Closed jiangxb98 closed 4 months ago
Sorry, this is a mistake of mine, I missed a space after the CUDA_VISIBLE_DEVICES='0,1,2,3', caused this mistake.
CUDA_VISIBLE_DEVICES='0,1,2,3'
CONFIG=$1 GPUS=$2 PORT=${PORT:-28509} CUDA_VISIBLE_DEVICES='0,1,2,3' \ PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \ python -m debugpy --listen 5678 --wait-for-client \ -m torch.distributed.launch --nproc_per_node=$GPUS --master_port=$PORT \ $(dirname "$0")/train.py $CONFIG --launcher pytorch ${@:3} --deterministic
Sorry, this is a mistake of mine, I missed a space after the
CUDA_VISIBLE_DEVICES='0,1,2,3', caused this mistake.