딥러닝/Hardware (1) 썸네일형 리스트형 [GPU] tensorflow gpu 선택하기 config = tf.ConfigProto() config.gpu_options.allow_growth = False [True 일 경우 필요할 때마다 GPU 사용률이 조절되고 False 는 GPU의 모든 메모리를 점유한다] config.gpu_options.visible_device_list = "1" [GPU 가 2개일 경우, 첫 번째 GPU만 사용하려면 "0", 두번째 GPU만 사용하려면 "1" 로 설정한다.] global_init = tf.global_variables_initializer() local_init = tf.local_variables_initializer() sess = tf.Session(config=config) sess.run([global_init, local_init]) 이전 1 다음