colab 에서 CUDA GPU 를 할당할 때, runtime error: no cuda gpus are available 오류가 발생하는 케이스가 있다.
우선적으로는 상단 메뉴에서 런타임 - 런타임 유형 변경 탭으로 진입하여 하드웨어 가속기가 GPU 로 설정되어 있는지 확인해야 한다.
위의 설정이 정상적으로 되어 있는 경우에도 오류가 발생한다면, 소스 상에 GPU 가 아래의 코드로 할당되어 있는지 확인해야 한다.
os.environ["CUDA_VISIBLE_DEVICES"]
해당 코드를 주석처리 해야 colab 에서 정상적으로 GPU 할당이 가능하다.
'Dev' 카테고리의 다른 글
[POSTMAN] Error: getaddrinfo ENOTFOUND (0) | 2023.02.15 |
---|---|
[Python] 라이브러리 import 오류 (0) | 2023.01.25 |
[colab] bash: Bad Substitution (0) | 2021.04.01 |
[colab] sh: Permission denied 오류 (0) | 2021.03.29 |
Metric Learning and Triplet Loss (0) | 2021.03.28 |
댓글