WarpFusion v0.14.13导入pytorch_lightning的seed_everything报错

1. seed_everything的正确位置 #

在适配WarpFusion v0.14.13的环境中，seed_everything的正确导入路径是pytorch_lightning.utilities.seed。另外，也可以用PyTorch原生API替代该函数的功能——torch.manual_seed()+torch.cuda.manual_seed_all()，配合Python标准库的random.seed()、numpy.random.seed()，同样能实现全局种子固定。

2. torchmetrics导入报错的解决方法 #

这个报错是因为torchmetrics与pytorch_lightning版本不兼容导致的，按以下步骤修复：

• 先卸载现有冲突版本：

  pip uninstall -y pytorch_lightning torchmetrics
  

• 安装与WarpFusion v0.14.13兼容的版本组合：

  pip install pytorch_lightning==1.9.5 torchmetrics==0.11.4
  

3. 不调整依赖的代码替换方案 #

如果暂时不想修改依赖版本，直接修改导入代码即可：

• 替换原错误导入：

  from pytorch_lightning import seed_everything
  

• 改为以下二选一：

  # 方案1：使用正确的PyTorch Lightning导入路径
  from pytorch_lightning.utilities.seed import seed_everything
  

  # 方案2：用原生API实现等价功能
  import torch
  import random
  import numpy as np
  
  def seed_everything(seed):
      torch.manual_seed(seed)
      torch.cuda.manual_seed_all(seed)
      random.seed(seed)
      np.random.seed(seed)
  

内容的提问来源于stack exchange，提问作者Faiz Satrio