from pytorch_lightning.utilities.distributed import rank_zero_only报错解决

最新推荐文章于 2024-03-22 16:13:02 发布

chari克里

最新推荐文章于 2024-03-22 16:13:02 发布

阅读量1.2k

点赞数 7

文章标签： pytorch 人工智能 python 报错

本文链接：https://blog.csdn.net/weixin_45615730/article/details/136871753

版权

本文介绍了在最新PyTorchLightning版本中，由于distributed包被拆分，如何解决因使用旧代码导致的reank_zero_only函数调用错误，需将引用替换为pytorch_lightning.utilities.rank_zero.rank_zero_only。

摘要由CSDN通过智能技术生成

No module named ‘pytorch_lightning.utilities.distributed’ 报错解决
在这里插入图片描述
解决：
将报错的代码替换为pytorch_lightning.utilities.rank_zero

from pytorch_lightning.utilities.rank_zero import rank_zero_only
#from pytorch_lightning.utilities.distributed import rank_zero_only

原因：
在最新版的pytorch_lightning中，取消了distributed包，将其中功能分散进其它包中。rank_zero_only函数移动至rank_zero包中。

关注