tensorflow eager gpu 할당 쓴만큼만 잡게하기.

그냥 돌리게 되면 한 노트북에서 쓰지도 않는 GPU를 다 잡아버린다.

아래 코드를 돌리게 되면 TEST 한다면서 GPU를 다 잡아서 다른 Notebook에서 할당이 안된다.

tf.test.is_gpu_available()

 

 

그래서 아래와 같은 방법으로 진행하면 된다.

 

그냥 Session으로 할 때

gpu_option = tf.GPUOptions(allow_growth=True)
config = tf.ConfigProto(log_device_placement=False ,
						allow_soft_placement=True,
						gpu_options=gpu_option)
sess =  tf.Session(config=config)

 

Eager로 할 때

import tensorflow as tf
gpu_option = tf.GPUOptions(allow_growth=True)
config = tf.ConfigProto(log_device_placement=False,
                        allow_soft_placement=True,
                        gpu_options=gpu_option)
tf.enable_eager_execution(config = config)