Link
10-20 13:08
«   2020/10   »
        1 2 3
4 5 6 7 8 9 10
11 12 13 14 15 16 17
18 19 20 21 22 23 24
25 26 27 28 29 30 31
Archives
Today
1,204
Total
1,074,685
관리 메뉴

꿈 많은 사람의 이야기

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

deep learning(딥러닝)

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

이수진의 블로그 이수진의 블로그 2018. 7. 22. 14:52


얼마전 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)


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

동작도 잘 되구요.


6 Comments
댓글쓰기 폼