module 'tensorflow' has no attribute 'GraphDef'

발생된 에러

예외가 발생했습니다. AttributeError
module 'tensorflow' has no attribute 'GraphDef'
  File "G:\office\enterprise_sdk_int_g_ui\develnote\deeplearning\cnn-colorize-master (1)\cnn-colorize-master\colorize.py", line 54, in main
    graph_def = tf.GraphDef()
  File "G:\office\enterprise_sdk_int_g_ui\develnote\deeplearning\cnn-colorize-master (1)\cnn-colorize-master\colorize.py", line 75, in <module>
    main()

 

원인은 tensorflow-gpu 가 설치되지 않아서 발생한다. pip 명령으로 tensorflow-gpu 를 설치한다.

 

또는 tensorflow 버전에 따라서 다를 수 있습니다.

 

tf.compat.v1.GraphDef()   # -> instead of tf.GraphDef()
tf.compat.v2.io.gfile.GFile()   # -> instead of tf.gfile.GFile()

tinyclouds.org/colorize/

Automatic Colorization