Google Colab 에서 GPU 환경 사용하기

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 '''

원문링크 : Google Colab 에서 GPU 환경 사용하기