Docker GPU 연결

이번엔 docker에서 gpu 사용가능하게 하는 방법을 포스팅하겠습니다.

저의 경우에는 GPU가 오픈 소스 커널 모듈(nvidia-open)만 지원이 가능해서 거기에 맞게 설치하였습니다.

1️⃣ 기존 NVIDIA 드라이버 완전 제거

sudo apt purge '^nvidia'
sudo apt autoremove --purge
sudo rm -rf /etc/X11/xorg.conf
sudo update-initramfs -u

2️⃣ 최신 NVIDIA 드라이버 (Open 모드 포함) 재설치

Ubudo ubuntu-drivers devices
sudo apt install nvidia-driver-550-open
sudo reboot
nvidia-smi

3️⃣ NVIDIA Container Toolkit 설치 여부 확인

dpkg -l | grep nvidia-container

• 없을경우 설치

distribution=$(. /etc/os-release;echo $ID$VERSION_ID) && \
sudo mkdir -p /etc/systemd/system/docker.service.d && \
curl -s -L https://nvidia.github.io/libnvidia-container/gpgkey | sudo apt-key add - && \
curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.list | \
  sudo tee /etc/apt/sources.list.d/nvidia-container.list

sudo apt update
sudo apt install -y nvidia-docker2
sudo systemctl restart docke```