About running on one GPU

dqshuai / MetaFormer

A PyTorch implementation of "MetaFormer: A Unified Meta Framework for Fine-Grained Recognition". A reference PyTorch implementation of “CoAtNet: Marrying Convolution and Attention for All Data Sizes”

MIT License

210 stars 36 forks source link

About running on one GPU #16

Open yuquanqingyin opened 2 years ago

yuquanqingyin commented 2 years ago

I have only one GPU. I have set local_rank=-1 ,but failed to run the code. What do I need to revise to successfully run on one GPU?

dqshuai commented 2 years ago

you can try "python3 -m torch.distributed.launch --nproc_per_node 1"?