■ 텐서플로 GPU 버전 실행시 세션을 설정하는 방법을 보여준다.
▶ 예제 코드 (PY)
|
1 2 3 4 5 6 7 8 9 10 11 12 |
import tensorflow as tf ... config = tf.ConfigProto() config.gpu_options.allow_growth = True with tf.Session(config = config) as session: ... |
■ 텐서플로 GPU 버전 실행시 세션을 설정하는 방법을 보여준다.
▶ 예제 코드 (PY)
|
1 2 3 4 5 6 7 8 9 10 11 12 |
import tensorflow as tf ... config = tf.ConfigProto() config.gpu_options.allow_growth = True with tf.Session(config = config) as session: ... |