在Google Colab训练StyleGAN2-ADA-PyTorch时遇Tensorboard导入错误

这个错误是因为新版Tensorboard移除了tensorboard.compat下的notf函数，和StyleGAN2-ADA-PyTorch的代码不兼容导致的，以下是两种可行解决方法：

方法一：降级Tensorboard到兼容版本 #

这是最直接的解决方案，执行以下命令在Colab中安装适配的Tensorboard版本：

pip uninstall -y tensorboard
pip install tensorboard==2.10.0

安装完成后重新运行你的训练命令即可。

方法二：修改代码中的导入逻辑 #

如果不想降级版本，可手动修改StyleGAN2-ADA-PyTorch代码中涉及notf的部分：

• 找到代码中包含from tensorboard.compat import notf的文件（通常在train.py或日志相关模块里）
• 将该行替换为：from tensorflow.python.util import deprecation
• 把代码中所有调用notf()的地方替换为deprecation.deprecated()，并调整参数匹配新函数的要求

额外注意 #

Colab默认的PyTorch版本若过高也可能引发兼容问题，可先检查当前版本：

import torch
print(torch.__version__)

推荐使用1.12.x系列版本，若版本不符可执行命令降级：

pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

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