我的环境:tensorflow_gpu == 1.15
报错原因是tensorflow-estimator版本不对。将tensorflow-estimator卸载了再重新装:
pip uninstall tensorflow_estimator
pip install tensorflow_estimator
安装完之后他提示安装的tensorflow_estimator和tensorflow不匹配,推荐安装1.15版本的。
根据提示再卸载、安装:
pip uninstall tensorflow_estimator
pip install tensorflow_estimator==1.15
再运行代码,还是报同样的错。
突然发现,还有个tensorflow-gpu-estimator
安装时,提示该包只有2.1.0, 2.2.0, 2.3.0, 2.4.0 等版本,我就选了个最原始的版本安装
pip uninstall tensorflow_gpu_estimator
pip install tensorflow_gpu_estimator==2.1.0
运行代码,发现不再报错!