tensorflow' has no attribute 'ConfigProto'

github.com/anishathalye/neural-style

상기 repositery에 tensorflow 소스를 빌드하는 도중 위와같은 오류가 발생하였다. tensorflow2.0, tenserflow 에서 나타는 에러이다. tensorflow에서는 ConfigProto() 또한 1.x 버전의 잔유물로 사용된다.

 

tf.compat.v1.ConfigProto() 로 사용해야 한다.

 

Session과 GPU를 잘 잡는지 확인하려고 보통 함께 사용하는 함수는 다음과 같다.

sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

 

상기에러를 수정하려면 아래와 같이 수정한다.

sess = tf.compat.v1.Session(config=tf.compat.v1.ConfigProto(log_device_placement=True))