728x90
config = tf.ConfigProto()
config.gpu_options.allow_growth = False
[True 일 경우 필요할 때마다 GPU 사용률이 조절되고 False 는 GPU의 모든 메모리를 점유한다]
config.gpu_options.visible_device_list = "1"
[GPU 가 2개일 경우, 첫 번째 GPU만 사용하려면 "0", 두번째 GPU만 사용하려면 "1" 로 설정한다.]
global_init = tf.global_variables_initializer()
local_init = tf.local_variables_initializer()
sess = tf.Session(config=config)
sess.run([global_init, local_init])
반응형