GCP로 GPU 사용하기

GCP란?
Google Cloud Platform의 약자로 Google이 제공하는 클라우드 컴퓨팅 서비스입니다.
여러가지 서비스를 제공하지만 그중에서도 GPU를 사용하는 방법을 포스팅하려 합니다.

 

0. GCP 로그인

• 처음 가입하면 40만원 상당의 무료 크레딧이 3개월간 제공됩니다. 

• GPU 할당량 요청

 I AM → GPUS-ALL-REGIONS-per-project 검색 후 선택  →  GPU 개수, 사유 입력 후 할당량 요청

 

1. Compute Engine에서 GPU 인스턴스 생성

1) 가상 머신(vm) 생성

• Compute Engine → vm  → 인스턴스 만들기 선택 

 

• 가상 머신 이름, 지역, 영역, gpu, cpu 선택 

 

2) os 및 스토리지

저는 이미지 전환 버튼을 눌러서 자동 설정 후 디스크 용량만 100으로 변경 해줬습니다.

 

3) 생성 완료

 

 

2. GPU 드라이버 설치

• NVIDIA 드라이버, CUDA 설치
• 위에서 생성한 VM 인스턴스에서 SSH 클릭하여 접속

→ Would you like to install the Nvidia driver? 이 나오면 y(yes) 입력 후 Enter

 

 

- 설치 확인

nvidia-smi

Driver, CUDA 모두 잘 설치된 것을 확인

 

• 추가로 필요한 라이브러리 설치

`pip list`로 현재 설치된 라이브러리들을 확인하고 추가로 필요한 라이브러리를 설치 해줍니다.