관리 메뉴
꿈 많은 사람의 이야기
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)
이 내용을 써주니까 에러가 사라졌습니다.
동작도 잘 되구요.
반응형
'deep learning(딥러닝)' 카테고리의 다른 글
cs231n 2017 3강 정리 loss functions and optimization (4) | 2018.10.31 |
---|---|
파이썬 케라스(keras) 딥러닝 CNN으로 강아지 종류(품종)을 예측해봅시다! (78) | 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 (81) | 2018.07.04 |
파이썬 케라스(keras)로 딥러닝하자! CNN을 이용해 이미지 분류하기(image classification) (160) | 2018.06.29 |
1st 함께하는 딥러닝 컨퍼런스를 갔다오다~ 너무 좋았다 (0) | 2018.06.28 |
- Tag
- deep learning, error, gpu error, gtx 1060, Keras, keras error, keras gpu, keras gpu error, nvidia, tensorflow gpu, tensorflow gpu error, 딥러닝, 에러, 케라스, 케라스 에러, 텐서플로우, 텐서플로우 gpu, 텐서플로우 에러
8 Comments
-
CUDNN 2019.05.16 17:51 감사합니다!! 덕분에 문제 해결했습니다ㅜㅜㅜ 제 은인이셔요....
-
오키여 이수진의 블로그 2019.05.16 22:47 신고 ㅎㅎ 도움 되셨다니 다행입니다! 화이팅!
-
jodonghee 2020.02.10 16:20 감사합니다 ㅋㅋㅋㅋ 덕분에 해결했네요
원리가 무엇인지 알 수 있을까요?? -
오키여 이수진의 블로그 2020.02.10 18:36 신고 도움 되셨다니 다행입니다 ㅎㅎ
해당 코드의 원리는 필요에 따라 GPU 메모리를 탄력적으로 사용한다고 합니다.~
-
comment 2020.06.11 10:40 저도 이런 에러가 계속 났었는데, 해결방법 알려주셔서 감사합니다!!
-
오키여 이수진의 블로그 2020.06.11 16:15 신고 도움 되셨다니 다행입니다 ㅎㅎ
-
정말감사합니다. 2021.02.14 13:50 덕분에 해결했습니다. 감사합니다.
-
오키여 이수진의 블로그 2021.02.16 08:51 신고 도움 되셨다니 다행입니다!