반응형
import tensorflow as tf
from keras.backend.tensorflow_backend import set_session
# tf.compat.v1.disable_eager_execution()
# config = tf.compat.v1.ConfigProto()
# config.gpu_options.per_process_gpu_memory_fraction = 0.1
# set_session(tf.compat.v1.Session(config=config))
config = tf.compat.v1.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.1
tf.compat.v1.keras.backend.set_session(tf.compat.v1.Session(config=config))
반응형
'연구개발 > Machine Learning' 카테고리의 다른 글
| [Chapter6] 모델 평가와 하이퍼파라미터 튜닝에 관한 사례학습 (0) | 2017.12.05 |
|---|---|
| [Chapter5] 차원축소를 이용한 데이터 압축 (0) | 2017.12.05 |
| [Chapter4] 데이터 전처리 (0) | 2017.12.04 |
| 모두를 위한 머신러닝과 딥러닝의 강의 (0) | 2017.03.13 |
| 7. 매트릭스와 벡터 계산하기(Linear algebra) (0) | 2017.03.03 |