使用VQGAN+CLIP Notebook遇rank_zero_only导入错误求助

第一步：确定pip命令的位置 #

在Colab里，你需要新建一个代码单元格，放在所有导入库的单元格最前面（也就是"Loading of libraries and definitions"这个单元格的上方）。操作方式：点击左上角的「+代码」按钮，然后把这个新单元格拖到目标单元格的最顶部。

第二步：安装兼容版本的依赖 #

在新建的单元格里粘贴以下命令，运行它：

!pip uninstall -y pytorch_lightning
!pip install pytorch_lightning==1.5.10
!pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

命令说明： #

• 卸载现有pytorch_lightning：你当前环境的版本和taming-transformers不兼容——1.6及以上版本的pytorch_lightning把rank_zero_only移到了pytorch_lightning.utilities.rank_zero模块，但旧版taming代码还是从distributed里导入，导致报错。
• 安装1.5.10版本的pytorch_lightning：这个版本和你用的VQGAN+CLIP notebook依赖的taming-transformers完全兼容。
• 重新安装PyTorch套件：确保PyTorch、torchvision版本匹配，适配Colab的GPU环境。

第三步：运行原单元格 #

等新单元格的安装命令执行完成后，再运行"Loading of libraries and definitions"单元格，就能正常导入库了。

内容的提问来源于stack exchange，提问作者Ines Sieulle