首页 > TAG信息列表 > accelerate
GPU多卡训练torch模型
用命令即可： python3 -m torch.distributed.launch --nproc_per_node 8 train.py 或者 使用Accelerator： https://huggingface.co/docs/transformers/v4.21.2/en/accelerateGPU多卡训练torch模型
用命令即可： python3 -m torch.distributed.launch --nproc_per_node 8 train.py 或者 使用Accelerator： https://huggingface.co/docs/transformers/v4.21.2/en/accelerate