完美解决Python中Numpy与TensorFlow版本兼容问题

import tensorflow as tf
import numpy as np
 
# 检查TensorFlow和Numpy版本是否兼容
def check_tf_np_versions():
    tf_version = tf.__version__
    np_version = np.__version__
 
    # 假设兼容性列表
    compatibility_list = {
        '2.6.0': '1.18',
        '2.7.0': '1.19',
        # 添加更多版本对应关系
    }
 
    # 检查TensorFlow版本是否在兼容性列表中
    if tf_version in compatibility_list:
        # 检查Numpy版本是否与TensorFlow版本兼容
        if np_version in compatibility_list[tf_version]:
            print(f"TensorFlow {tf_version} is compatible with Numpy {np_version}.")
        else:
            print(f"Warning: Numpy {np_version} is not compatible with TensorFlow {tf_version}.")
            print(f"You should upgrade Numpy to {compatibility_list[tf_version]}.")
    else:
        print(f"Warning: TensorFlow {tf_version} is not on the compatibility list.")
        print("Please check the TensorFlow and Numpy compatibility list.")
 
check_tf_np_versions()

这段代码定义了一个函数check_tf_np_versions，它检查TensorFlow和Numpy的版本是否兼容。它使用一个字典compatibility_list来存储已知的兼容版本对，并给出相应的警告或提示。这样的方法可以有效地处理版本兼容性问题，并在开发者需要时提供升级建议。