GPU 메모리 부족

nvidia-smi 로 power shell에서 확인해보니

6003MiB / 6144 MiB

메모리가 부족한 걸 확인할 수 있었다.

 

켜져있던 게 뭐가 많았던 모양

 

 

 

 

 

 

 

 

 

1. 프로그램을 모두 꺼주고 다시 확인

539MiB / 6144 MiB

확실히 떨어졌다

 

 

 

 

 

 

 

2. 그럼에도 다시 파이썬 코드를 실행시키면 같은 에러가 발생한다면

 

 

GPU 용량 제한을 걸어주자

import tensorflow as tf
from keras.backend.tensorflow_backend import set_session
from tensorflow.python.keras import backend as K

config = tf.compat.v1.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.3
K.set_session(tf.compat.v1.Session(config=config))