[ErrorBook] CUDA GPU

hyewon·2022년 3월 8일
0

ErrorBook

목록 보기
3/3

colab을 기준으로 작성되었습니다.

🎃 오늘의 에러

RuntimeError: no CUDA GPUs are available

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

딥러닝 모델을 학습시키는 도중에 다음과 같은 에러가 떴다.

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

그리고 설치를 한 뒤에 꼭! 런타임 재시작을 해줘야한다. 해주지 않고 바로 에러났던 코드를 다시 실행하면 아래와 같은 에러가 난다😂

아래 에러에서 헤메다가 런타임 재시작을 해주니까 에러가 하나도 안났다. 뒷걸음질 치다가 해결했지만 에러는 해결하긴 했다...!

profile
우당탕탕 코린이

0개의 댓글