How to run on multiple GPUs?

VISION-SJTU / RECCE

[CVPR2022] End-to-End Reconstruction-Classification Learning for Face Forgery Detection

MIT License

116 stars 16 forks source link

How to run on multiple GPUs? #40

Closed CharlesTran closed 1 month ago

XJay18 commented 3 months ago

Hi, it has been a long time since the code deployment, and I currently do not have a similar environment for this project. However, I think you may try the following command first. Please also refer to PyTorch Distributed Overview for multi-GPU training.

CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 --master_port 12345 train.py --config path/to/config.yaml

XJay18 commented 1 month ago

Close due to inactivation. Please feel free to reopen this issue if you still have related problems.