[에러 해결] CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling `cublasCreate(handle)`

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

사전학습된 모델과 토크나이저를 불러오는 코드 부분을 수정했더니 위와 같은 에러가 발생했다.

원인은 토크나이저에 새로운 스페셜 토큰을 추가했는데, 모델에 대해 resize 해주는걸 까먹지 말아야지 해놓고 까먹었기 때문이다. ㅠ

 

해결 방법

model.resize_token_embeddings(len(tokenizer))

토크나이저와 모델을 선언한 다음 이 코드를 추가해준다.