Train with 3 GPUs
yunzhelimac opened this issue · 3 comments
yunzhelimac commented
您好，
我想问一下要是分三张卡跑该怎么修改代码呢
简单地将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的代码在哪里呢
谢谢
Hihermon commented
你好，请问您的问题解决了吗
dzw007 commented
你好，请问您的问题解决了吗
dzw007 commented
两张卡--nproc_per_node改成2可以