背景:pytorch 2.0.0
我定义的是local_rank,而报错显示的是local-rank。这不是我项目的问题。搜了一下发现应该是torch版本问题。
有看到其他方法说: 降低版本为1.13 可以解决。但是我不想降版本。搜了一下也解决了。
torchrun 替换掉python -m torch.distributed.launch
通过这种方法解决了这一问题。
背景:pytorch 2.0.0
我定义的是local_rank,而报错显示的是local-rank。这不是我项目的问题。搜了一下发现应该是torch版本问题。
有看到其他方法说: 降低版本为1.13 可以解决。但是我不想降版本。搜了一下也解决了。
torchrun 替换掉python -m torch.distributed.launch
通过这种方法解决了这一问题。