keras, tensorflow gpu 에러(failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED, could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED)

얼마전 DELL 최신 노트북으로 바꾸면서 tensorflow-gpu를 설치했습니다.

그리고 keras(케라스)를 설치했지요

즉 백엔드로 텐서플로우를 사용하는 케라스를 사용했습니다.

하지만 여러 문제점이 조금 있었습니다.

failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED

could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED

이런 에러들이 뜨면서 실행이 되지 않았는데요.

구글링을 해보니 저 뿐만 아니라 다른 여러 사람들도 겪었던 문제네요.

정확한 원인은 모르겠지만, gpu 메모리 할당 문제라고 합니다.

그 결과 원인을 해결할 수 있는 줄을 추가했습니다.

저는 윈도우10 환경에서 tensorflow-gpu를 설치했구요. 그 위에서 keras를 돌렸습니다.

이렇게 에러가 뜨면

위와 같은 소스를 추가해주시면 됩니다.

저는 케라스(keras)가 backend로 tensorflow를 돌리고 있습니다.

윈도우 10 환경 기준이구요

import tensorflow as tf

config = tf.ConfigProto()

config.gpu_options.allow_growth = True

session = tf.Session(config=config)

이 내용을 써주니까 에러가 사라졌습니다.

동작도 잘 되구요.