FutureWarning: The module torch.distributed.launch is deprecated and will be removed in future. Use torchrun. Note that --use-env is set by default in torchrun. If your script expects `--local-rank` argument to be set, please change it to read from `os.environ['LOCAL_RANK']` instead. See https://pytorch.org/docs/stable/distributed.html#launch-utility for further instructions
时间: 2024-04-09 20:32:23 浏览: 292
这个警告提示表示 torch.distributed.launch 模块已被弃,并将在未来的中删除。建议 torchrun 替代它。
使用 torchrun 时,默认情况下,--use-env 选项已经被设置好了。如果你的脚本期设置 --local-rank 参数,请为从 os.environ[''] 中读取该值。可以在这里参考更多的指令:https://pytorch.org/docs/stable/distributed.html#launch-utility
建议你更新你的代码,使用 torchrun 来启动分布式训练,并根据警告信息中的指引来进行相应的修改。
如果你有任何其他问题,请随时提问。
阅读全文