딥 러닝을 위한 Google Colab 설정

소개

Google Colaboratory 또는 Colab은 시스템에 Python을 설치하지 않고도 Python 노트북을 실행할 수 있도록 Google에서 제공하는 무료 서비스입니다. Colab은 사용자에게 다양한 기능을 제공하며 그 중 가장 중요한 기능은 다음과 같습니다.

기본 노트북에 대해 구성할 필요 없음

자유롭고 간편한 GPU 액세스

쉽게 코드를 공유하고 노트북을 GitHub에 연결

Google 드라이브에서 바로 데이터 세트를 로드하고 노트북/훈련된 모델을 드라이브에도 저장합니다.
Google Colab에는 Pro 기능도 함께 제공됩니다. Colab Pro에 대한 자세한 내용은 here에서 확인할 수 있습니다.
이 블로그 게시물에서는 기계 학습 또는 딥 러닝 개발을 시작하기 위해 Colab 노트북을 설정하는 방법에 대해 설명합니다.

종속성 설치

Colab은 일반적으로 Tensorflow, PyTorch, scikit-learn, pandas 등과 같은 대부분의 기본 종속 항목과 함께 사전 설치되어 제공되지만 때때로 외부 패키지를 설치해야 할 가능성이 있습니다. !pip install 명령을 사용하여 이를 수행할 수 있습니다. 예를 들어 테스트 단계에서 이미지를 보강하는 데 사용되는 ttach 라이브러리를 설치할 수 있습니다. 이것은 다음을 사용하여 수행할 수 있습니다.
!pip install library-name
셀을 실행하면 다음과 같은 결과가 출력됩니다.

이렇게 하면 필요에 따라 사용자 정의 라이브러리를 설치할 수 있습니다.

Google 드라이브 마운트

아래 제공된 간단한 스크립트를 사용하여 Google 드라이브를 마운트할 수 있습니다.

from google.colab import drive
drive.mount('/content/drive')

이것을 실행하면 마운트하려는 Google 드라이브의 계정을 선택하도록 리디렉션하는 링크가 표시됩니다. 계정을 선택하고 권한 요청을 수락한 후 화면에 나타나는 코드를 복사하여 대화 상자에 붙여넣고 Enter 키를 누릅니다. 이제 Google 드라이브가 마운트되었으며 왼쪽의 파일 메뉴를 통해 액세스할 수 있습니다. 출력은 다음과 같습니다.

GitHub 리포지토리 복제

때때로 여러 스크립트를 활용하는 복잡한 프로젝트에서 작업하기 위해 GitHub 리포지토리를 Colab 라이브러리에 복제해야 할 수도 있습니다. !git clone link-to-repository를 사용하여 수행할 수 있습니다.
셀을 실행하면 리포지토리가 작업 디렉터리에 복제됩니다.

리포지토리를 마운트한 경우 Google 드라이브에 리포지토리를 복제할 수도 있습니다. !git clone link-to-repository path-to-drive 를 사용하면 됩니다. 샘플 출력은 다음과 같습니다.

GPU 런타임에 액세스

Colab의 최고 기능 중 하나는 Colab이 제공하는 무료 GPU 런타임입니다. 다음을 통해 GPU 런타임을 활성화할 수 있습니다.

런타임으로 이동

런타임 유형 변경을 선택합니다.

드롭다운 메뉴에서 GPU를 선택하고 저장을 클릭합니다.

이제 Colab에서 제공하는 GPU 런타임을 사용할 수 있습니다.!nvidia-smi를 사용하여 제공된 GPU를 확인할 수도 있습니다.
출력에는 사용 중인 GPU 메모리와 기타 세부 정보도 표시됩니다.

이것으로 기계 학습 및 딥 러닝 애플리케이션을 실행하기 위한 Colab 설정에 대한 논의를 마칩니다. 이제 GitHub 리포지토리를 드라이브에 쉽게 복제하고 로컬 컴퓨터에 Python을 설치하지 않고도 GPU 런타임을 사용하여 실행할 수 있습니다.