Development/Environment

Google Colab 에서 GPU 환경 사용하기

고고마코드 2021. 12. 17. 23:45
반응형

Google Colab 을 사용할 때 기본으로는 CPU 환경만 사용할 수 있기에 GPU 환경을 사용하기 위해서는 설정 변경을 해야 한다.


Google Colab 에서 GPU 환경 세팅하기

Google Colab 실행

자신의 드라이브에서 Google Colaboratory 을 실행한다.

런타임 유형 변경

상단 목록에서 [런타임 > 런타임 유형 변경] 을 선택

하드웨어 가속기 [None -> GPU] 로 변경

GPU 설정 완료


GPU 환경 테스트

import torch
print(torch.cuda.is_available())
'''
  CPU 환경일 경우: False
  GPU 환경일 경우: True
'''
반응형