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)

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

동작도 잘 되구요.

반응형

그리드형

저작자표시

'deep learning(딥러닝)' 카테고리의 다른 글

cs231n 2017 3강 정리 loss functions and optimization  (4)	2018.10.31
파이썬 케라스(keras) 딥러닝 CNN으로 강아지 종류(품종)을 예측해봅시다!  (94)	2018.10.01
keras, tensorflow gpu 에러(failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED, could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED)  (8)	2018.07.22
파이썬 케라스(keras)로CNN 딥러닝하자! cat dog binary image classification  (85)	2018.07.04
파이썬 케라스(keras)로 딥러닝하자! CNN을 이용해 이미지 분류하기(image classification)  (197)	2018.06.29
1st 함께하는 딥러닝 컨퍼런스를 갔다오다~ 너무 좋았다  (0)	2018.06.28