pip install accelerate

初始化配置:accelerate config

或者使用默认配置:accelerate config default

CUDA_VISIBLE_DEVICES=0,2,3,5 accelerate launch --num_processes=4 --main_process_port 29501 train_dreambooth.py

  • 默认端口为29500,同时运行多个程序需要指定端口
  • 设置CUDA_VISIBLE_DEVICES和num_processes来选择使用的显卡,否则将全部使用
Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐