Programming/errors

[CUDA error] : CUBLAS_STATUS_ALLOC_FAILED

방황하는 데이터불도저 2022. 11. 10. 12:38

CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`

 

(jupyter lab 환경에서 작업)

 

구글링을 통해 얻은 해결방법

1) 배치사이즈를 줄이고, 코드를 다시 restart시키면 작동하는 것을 확인했다.

    - 하지만 기존에 잘 돌아가던 코드에서 갑자기 저런 에러가 발생하여 당황스럽다...

 

2) 배치사이즈를 줄여도 해결되지 않는 경우도 있다.

     (사실 저 에러는 배치사이즈와 관련이 없다는 사실)

https://stackoverflow.com/questions/61473330/cuda-error-cublas-status-alloc-failed-when-calling-cublascreatehandle