参考：如何在pycharm中运行/调试torch分布式训练 - 知乎

1 建立软件

首先需要找到torch.distributed.launch这个文件，并将它软链接到我们的Pycharm项目目录下。

ln -s /opt/anaconda3/envs/swin/lib/python3.8/site-packages/torch/distributed/ /home/ubuntu/wgp/Swin-Transformer/

2 设置pycharm

打开Pycharm，依次点击Run->Edit Configurations 进入参数配置界面,修改script path和parameters。