背景

使用 accelerate 进行分布式代码训练时，需要在pycharm中进行调试，此时需要在pycharm中运行。

终端执行命令

# *[Specify the config file path and the GPU devices to use]
export CUDA_VISIBLE_DEVICES=0# *[Specify the config file path]
export OMINI_CONFIG=./train/config/subject_dj.yamlecho $OMINI_CONFIG
export TOKENIZERS_PARALLELISM=true# 这里是使用单卡进行训练
accelerate launch --num_processes=1 --num_machines=1 --machine_rank=0 --main_process_port 41353 -m omini.train_flux.train_subject_dj

Pycharm设置

参考

1, accelerate在pycham中执行的方法_pycharm accelerate-CSDN博客

2, accelerate在pycham中执行的方法_pycharm accelerate-CSDN博客

3, https://github.com/huggingface/accelerate/issues/535