728x90
[tensorboard]
아나콘다 명령 prompt >tensorboard --logdir=./path/logs/
[gpu 메모리 관리] tf version 1.xx
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config)
[gpu 메모리 관리] tf version 2.xx
config = tf.compat.v1.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.compat.v1.Session(config=config)
[gpu 사용량 80%]
config = tf.compat.v1.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.8
session = tf.compat.v1.Session(config=config)
반응형
'👾 Deep Learning' 카테고리의 다른 글
Optimizer ( Adam, SGD ) (0) | 2021.03.27 |
---|---|
손실 함수 (loss function) (0) | 2021.03.07 |
OSError: [WinError 127] 지정된 프로시저를 찾을 수 없습니다. Error loading \\torch\\lib\\*_ops_gpu.dll or one of its dependencies. (0) | 2021.03.06 |
TFBertModel parameter (0) | 2021.03.05 |
Softmax RuntimeWarning 해결 (0) | 2021.03.03 |