Question

无论我做什么，我似乎都无法使TensorFlow-gpu与textgenrnn很好地配合。 Tensorflow-gpu测试表明一切正常运行目前，我收到错误消息：

ModuleNotFoundError: No module named 'tensorflow.contrib'

当我尝试将textgenrnn导入为

textgen = textgenrnn(name=model_name)

有人可以提供他们的工作开发环境吗？我的点子清单是：

Package              Version
-------------------- -------------------
absl-py              0.8.1
astor                0.8.0
bleach               1.5.0
cachetools           3.1.1
certifi              2019.9.11
chardet              3.0.4
gast                 0.2.2
google-auth          1.6.3
google-auth-oauthlib 0.4.1
google-pasta         0.1.7
grpcio               1.24.3
h5py                 2.10.0
html5lib             0.9999999
idna                 2.8
joblib               0.14.0
Keras                2.3.1
Keras-Applications   1.0.8
Keras-Preprocessing  1.1.0
Markdown             3.1.1
numpy                1.17.3
oauthlib             3.1.0
opt-einsum           3.1.0
pip                  19.3.1
protobuf             3.10.0
pyasn1               0.4.7
pyasn1-modules       0.2.7
PyYAML               5.1.2
requests             2.22.0
requests-oauthlib    1.2.0
rsa                  4.0
scikit-learn         0.21.3
scipy                1.3.1
setuptools           41.5.1
six                  1.12.0
tb-nightly           2.1.0a20191029
tensorboard          2.0.1
tensorflow-estimator 2.0.1
tensorflow-gpu       2.0.0
termcolor            1.1.0
textgenrnn           1.5.0
tf-estimator-nightly 2.0.0.dev2019102908
tqdm                 4.36.1
urllib3              1.25.6
Werkzeug             0.16.0
wheel                0.33.6
wrapt                1.11.2

我的Python版本是3.6.4，CUDA 10和cudnn 7.4.1.5。我尝试使用较低/较高的python版本以及CUDA 9。谢谢。

Answer 1

谈到黑金恶魔，几天后我终于明白了，尽管有控制台警告。

完成上述配置，然后卸载tensorflow-gpu以安装tensorflow-gpu == 1.14.0。

我希望这可以帮助某人。

Answer 2

张量流版本似乎有问题。作为contrib模块，在TensorFlow 2.0中不再可用您如何再次安装较低版本的tensorflow-gpu。我更喜欢v1.14

pip3 instal tensorflow-gpu==1.14

Windows 10上的Tensorflow-gpu和Textgenrnn版本不兼容：ModuleNotFoundError：没有名为'tensorflow.contrib'的模块

2 个答案: