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)

 
반응형
다했다