이전 글에 있는 정도의 환경설정만 하더라도 GPU를 충분히 활용할 수 있을 것이다. 그러나 필자의 경우에는 주로 Docker 환경에서 작업하는 것을 선호하는 편이다. 그래서 이번 편에는 Docker와 Nvidia-Docker을 설치하는 법을 다루고자 한다.

Docker 설치

Docker 설치는 사실 공식 문서를 보고 진행하는 것이 가장 편하다. 아래 링크로 들어가서 천천히 따라 설치해보도록 하자.

https://docs.docker.com/engine/install/ubuntu/

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 설치
sudo apt-get install -y nvidia-docker2

# docker 재시작
sudo systemctl restart docker

# 제대로 설치되었는지 확인
sudo docker run --rm --gpus all nvidia/cuda:11.8-base nvidia-smi