Deeplearning

RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED ...

by OBG posted Sep 06, 2022
?

단축키

Prev이전 문서

Next다음 문서

ESC닫기

크게 작게 위로 아래로 게시글 수정 내역 댓글로 가기 인쇄

https://discuss.pytorch.org/t/runtimeerror-cuda-error-cublas-status-alloc-failed-when-calling-cublascreate-handle/78545/13

 

위 링크 중 답변

 

The error might also be raised, if you are running out of memory on the GPU and cublas is unable to create the handle.
Could you reduce the batch size and check, if the code is working?

 

CPU 로 돌렸을 때 문제 없다면 모델이 큰 경우일 확률 높음.