■ 텐서플로 GPU 버전 실행시 세션 설정하기

------------------------------------------------------------------------------------------------------------------------

import tensorflow as tf

 

...

 

config = tf.ConfigProto()

 

config.gpu_options.allow_growth = True

 

with tf.Session(config = config) as session:

...

------------------------------------------------------------------------------------------------------------------------

Posted by 사용자 icodebroker