力所能及的改变这个世界 自驾相关碎碎念
搜索
文章 20

强化学习

解决由于layers引起的“ No module named ‘tensorflow.contrib‘“

12 月 9, 2024Jamil1 min
返回基地

引入以下包时:

import tensorflow.contrib.layers as tf_layers

报错:
ModuleNotFoundError: No module named 'tensorflow.contrib'

问题原因:

TensorFlow2.0删除已经删除tensorflow.contrib包

解决方法:

方法一:TensorFlow降级

pip uninstall tensorflow         
pip install tensorflow==1.14.0 

由于我的python环境为python3.6,提示没有合适的版本。

方法二:TF-slim包

pip install --upgrade tf-slim
讲引入的包改为:
import tf_slim.layers as tf_layers

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注