colab을 기준으로 작성되었습니다.
RuntimeError: No CUDA GPUs are available
device = torch.device("cuda")
.
.
.
model = BERTClassifier(bertmodel, dr_rate=0.5).to(device)
위의 코드를 실행하는데 CUDA GPU 에러가 떴다.
런타임 -> 런타임 유형 변경 -> 하드웨어 가속기 GPU 설정
내 경우에는 colab에서 GPU를 처음 사용했기 때문에 GPU가 None으로 되어있었다. 그래서 사용할 수 있는 GPU가 없다고 에러가 났다. 하드웨어 가속기를 GPU로 설정한 뒤 다시 코드를 실행시켜주면 에러 없이 잘 돌아간다 !
딥러닝 모델을 학습시키는 도중에 다음과 같은 에러가 떴다.
RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling
cublasCreate(handle)
해당 에러는 굉장히 간단하게 해결했는데 바로 pytorch 버전과 cuda 버전을 맞춰주는 것이다.
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
그리고 설치를 한 뒤에 꼭! 런타임 재시작을 해줘야한다. 해주지 않고 바로 에러났던 코드를 다시 실행하면 아래와 같은 에러가 난다😂
아래 에러에서 헤메다가 런타임 재시작을 해주니까 에러가 하나도 안났다. 뒷걸음질 치다가 해결했지만 에러는 해결하긴 했다...!