accelerate教程
或者使用默认配置:accelerate config default。初始化配置:accelerate config。
·
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来选择使用的显卡,否则将全部使用
更多推荐


所有评论(0)