AttributeError: module 'tensorflow._api.v1.compat.v1' has no attribute 'contrib'

Question

発生している問題・エラーメッセージ

Traceback (most recent call last):
File "run_classifier.py", line 1027, in <module>
tf.app.run()
File "/opt/conda/envs/cubert_v2/lib/python3.7/site-packages/tensorflow_core/python/platform/app.py", line 40, in run
_run(main=main, argv=argv, flags_parser=_parse_flags_tolerate_undef)
File "/opt/conda/envs/cubert_v2/lib/python3.7/site-packages/absl/app.py", line 303, in run
_run_main(main, args)
File "/opt/conda/envs/cubert_v2/lib/python3.7/site-packages/absl/app.py", line 251, in _run_main
sys.exit(main(argv))
File "run_classifier.py", line 871, in main
is_per_host = tf.contrib.tpu.InputPipelineConfig.PER_HOST_V2
File "/opt/conda/envs/cubert_v2/lib/python3.7/site-packages/tensorflow_core/python/util/module_wrapper.py", line 193, in getattr
attr = getattr(self._tfmw_wrapped_module, name)
AttributeError: module 'tensorflow._api.v1.compat.v1' has no attribute 'contrib'

のエラーが出て困っています。

私が調べた限り、解決方法は
1.tensorflow1系に切り替える
2.tf.disable_v2_behavior()を追加する
の２つがあることがわかりました。

どちらも試したのですが直らないので困っています。

Comments

[jbpb0]

エラーは省略しないで全部書いてください
「Traceback」と書かれてたら、そこから全部

[jbpb0]

> 1.tensorflow1系に切り替える

の場合ですが、

> module 'tensorflow._api.v1.compat.v1' has no attribute 'contrib'

から類推すると、もしかしたらTF 1.*で

import tensorflow.compat.v1 as tf

みたいなことしてませんでしょうか？
もしそうしてたら、上記はTF 2.*でやる方法なので、TF 1.*の場合は普通に

import tensorflow as tf

としてみてください

[jbpb0]

TF 2.*で

> 2.tf.disable_v2_behavior()を追加する

の場合ですが、

> module 'tensorflow._api.v1.compat.v1' has no attribute 'contrib'

http://blog.livedoor.jp/tak_tak0/archives/52431380.html
を見てください

[kaibe1002]

まさに、import tensorflow.compat.v1 as tfをしていました。
import tensorflow as tf　に変更するとこちらのエラーは解消できました。
ありがとうございました。

Answer 1

1.tensorflow1系に切り替える

の場合は、

import tensorflow.compat.v1 as tf

としないで、

import tensorflow as tf

としてみてください

　

TF 2を使う場合は、

module 'tensorflow._api.v1.compat.v1' has no attribute 'contrib'

BERTオフィシャルコードをtensorflow2で実行する方法
を見てください