环境:macbook上 tensorflow2.0 CPU版本
在使用 from tensorflow.contrib import rnn 时会报:“ModuleNotFoundError: No module named 'tensorflow.contrib' ”
原因是TF2.0 已经移除contrib 模块了,想要使用这个模块需要将TF降到TF1.x版本。
步骤:
1.查看本机安装的版本:
import tensorflow as tf
tf.__version__ # 输出2.0.0
2. 查看TF是CPU 版本还是GPU版本, 发现是CPU版本
import os
from tensorflow.python.client import device_lib
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "99"
if __name__ == "__main__":
print(device_lib.list_local_devices())
3.卸载TF2.0 pip uninstall tensorflow 并且移除相关文件
rm -r /Users/qfikh/anaconda3/lib/python3.7/site-packages/tensor*
3.再指定你要安装的tensorflow版本,比如我要的就是1.15.0
pip install tensorflow==1.15.0