[PYTHON/TENSORFLOW] 텐서플로 GPU 버전 실행시 세션 설정하기

icodebroker TENSORFLOW 2024-06-18

■ 텐서플로 GPU 버전 실행시 세션을 설정하는 방법을 보여준다.

▶ 예제 코드 (PY)


import tensorflow as tf

...

config = tf.ConfigProto()

config.gpu_options.allow_growth = True

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

import tensorflow as tf

...

config = tf.ConfigProto()

config.gpu_options.allow_growth = True

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

...

Post Views: 13