NAFNet icon indicating copy to clipboard operation
NAFNet copied to clipboard

Published 20 hours ago • verified publisher icon megvii-research

Train with 3 GPUs

Open yunzhelimac opened this issue 2 years ago • 3 comments

您好，

我想问一下要是分三张卡跑该怎么修改代码呢

简单地将python -m torch.distributed.launch --nproc_per_node=8 --master_port=4321 basicsr/train.py -opt options/train/NAFSSR/NAFSSR-B_x4.yml --launcher pytorch里的--nproc_per_node改成3貌似不行

想问一下您具体的enable multiple gpus on torch的代码在哪里呢

谢谢

yunzhelimac avatar Feb 08 '23 15:02 yunzhelimac

你好，请问您的问题解决了吗

Hihermon avatar Apr 10 '23 12:04 Hihermon

你好，请问您的问题解决了吗

dzw007 avatar May 18 '23 04:05 dzw007

两张卡--nproc_per_node改成2可以

dzw007 avatar Jun 12 '23 10:06 dzw007