GPU 런타임을 사용하는 Colab 노트북 설정

Colab 노트북을 생성하면

Google Colab
; https://colab.research.google.com/

무조건 GPU 환경이 되는 것은 아닙니다. 확인을 위해 다음과 같이 간단한 스크립트를 실행할 수 있는데요,

import torch

device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
print(device)

출력이 "cpu"로 나오면 그것은 GPU를 사용하지 않는 실행 환경임을 의미합니다. 위의 스크립트가 아니어도 간단하게 확인하는 방법이 있는데요, Colab 화면에서 "View resources"를 선택했을 때,



이렇게 "System RAM"과 "Disk"만 나오면 역시나 GPU 실행환경이 아닙니다.



현재의 런타임 환경을 GPU 추가로 바꾸고 싶다면, 위의 "Resources" 뷰 하단에 있는 "Change runtime type" 링크를 선택해,



다음과 같이 "Hardware accelerator"를 "GPU"로 바꿔줘야 합니다.



그래서 다시 스크립트를 실행하면 "cuda"가 나올 것이고, "Resources" 뷰에서도 "GPU RAM" 항목이 추가된 것을 볼 수 있습니다.

---

[이 글에 대해서 여러분들과 의견을 공유하고 싶습니다. 틀리거나 미흡한 부분 또는 의문 사항이 있으시면 언제든 댓글 남겨주십시오.]

[최초 등록일: 6/26/2023]
[최종 수정일: 6/26/2023]


이 저작물은 크리에이티브 커먼즈 코리아 저작자표시-비영리-변경금지 2.0 대한민국 라이센스에 따라 이용하실 수 있습니다.

by SeongTae Jeong, mailto:techsharer at outlook.com