在Jupyter Notebook运行StyleGAN报tensorflow无Dimension属性错如何解决

你遇到的AttributeError: module 'tensorflow' has no attribute 'Dimension'错误是版本不兼容导致的：官方原版StyleGAN及配套的encoder代码均基于TensorFlow 1.x开发，tf.Dimension是TensorFlow 1.x的专属API，该接口在TensorFlow 2.x版本中已被重构移除，你当前环境安装的是TensorFlow 2.x版本，所以调用时会抛出属性不存在的报错。

以下三种方案可任选其一解决问题：

• 方案1：降级TensorFlow到1.x适配版本
  StyleGAN官方推荐使用TensorFlow 1.14/1.15版本运行代码，执行对应安装命令即可：
  GPU版：pip install tensorflow-gpu==1.15
  CPU版：pip install tensorflow==1.15
  注意TensorFlow 1.15最高支持Python 3.7版本，需确保你的Python版本不高于3.7，否则会安装失败。
• 方案2：手动修改报错代码兼容TF2.x
  找到报错堆栈中提到的dnnlib/tflib/tfutil.py文件，将第34行中的tf.Dimension替换为tf.compat.v1.Dimension即可解决当前报错。如果后续运行还有其他TensorFlow API相关报错，统一将对应tf.xxx写法替换为tf.compat.v1.xxx，也可以在代码入口文件开头加入tf.compat.v1.disable_v2_behavior()关闭TensorFlow 2.x的默认行为，提升整体兼容性。
• 方案3：使用官方迁移工具批量适配TF2.x
  直接在项目根目录执行迁移命令：tf_upgrade_v2 --intree ./ --outtree ./stylegan_tf2 --reportfile migrate_report.txt，工具会自动扫描所有代码文件，将TensorFlow 1.x的API批量替换为2.x兼容的写法，迁移完成后直接使用stylegan_tf2目录下的代码运行即可。

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