[Tensorflow/에러 해결] Memory growth cannot differ between GPU devices

출처 : https://fixexception.com/tensorflow/memory-growth-cannot-differ-between-gpu-devices/

 

메모리 증가를 허용을 결정하는 함수인 tf.config.experimental.set_memory_growth를 이용할 때 다음과 같은 에러가 발생할 수 있다.

 

ValueError: Memory growth cannot differ between GPU devices

 

이는 gpu가 여러 개 있을 때, 모든 gpu에 대해 memory growth를 동일하게 세팅해주지 않아서 발생하는 에러이다.

예를 들어 gpu가 두 대 있는데 하나의 gpu에만 set_memory_growth를 True로 설정해주었을 경우이다.

이 경우에는 모든 gpu에 대해 set_memory_growth를 동일하게 설정해주면 된다.

 

예시)

gpus = tf.config.list_physical_devices('GPU')
for i in range(len(gpus)):
	tf.config.experimental.set_memory_growth(gpu[i], True)