Docker 컨테이너에서 GPU 사용 설정하기

Docker 컨테이너에서 GPU 사용 설정하기

Docker를 사용하면 애플리케이션을 컨테이너화하여 쉽게 배포하고 실행할 수 있습니다. 특히 딥러닝이나 머신러닝 작업을 수행할 때 GPU를 활용하면 성능이 크게 향상됩니다. 이번 포스트에서는 Docker 컨테이너에서 GPU를 사용할 수 있도록 설정하는 방법을 알아보겠습니다.

1. NVIDIA Docker 설치

먼저, NVIDIA Docker가 설치되어 있어야 합니다. NVIDIA Docker는 GPU 가속을 사용하는 Docker 컨테이너를 실행할 수 있도록 도와줍니다. 설치하려면 다음 명령어를 실행하세요:

# NVIDIA Docker 패키지 저장소 추가
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list

# 패키지 목록 업데이트
sudo apt-get update

# nvidia-docker2 패키지 설치 및 Docker 데몬 설정 재로드
sudo apt-get install -y nvidia-docker2
sudo systemctl restart docker

이 후 Docker 데몬을 재시작합니다:

sudo systemctl restart docker

2. Docker 스크립트 작성

GPU를 사용하는 Docker 컨테이너를 실행하는 스크립트를 작성하고 실행합니다.

Docker 컨테이너에서 GPU 사용 설정 완료

이제 Docker 컨테이너에서 GPU를 사용할 수 있도록 설정하는 방법을 배웠습니다. 이러한 설정을 통해 딥러닝 모델 학습과 같은 작업에서 GPU의 성능을 최대한 활용할 수 있습니다.