본문 바로가기

Dev Platform/Docker

[Docker] nvidia docker 및 nvidia/cuda 이미지 설치하기

반응형

안녕하세요.

Docker에서 GPU를 사용할 수 있는 방법을 알아보겠습니다.

https://github.com/NVIDIA/nvidia-docker

 

NVIDIA/nvidia-docker

Build and run Docker containers leveraging NVIDIA GPUs - NVIDIA/nvidia-docker

github.com

위 페이지를 참고하였습니다.

위 페이지 내용을 요약하면 nvidia-docker2를 설치하고 nvidia-docker로 docker을 실행합니다.

1. nvidia-docker 설치

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \

sudo apt-key add -

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)

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

sudo apt-get install -y nvidia-docker2

sudo pkill -SIGHUP dockerd

1 ~ 2 : 패키지 repositories를 추가합니다.

3 ~ 5 : 자신의 OS 버전에 맞게 nvidia docker 설치를 위한 소스 리스트를 추가합니다.

제 PC의 OS는 우분투 18.04입니다.

본문에는 docker.io에서 설치가 된다고 했지만... 저같은 경우에 docker.ce를 설치해야한다고 했습니다.

그래서 기존 docker.io를 제거하고 docker.ce로 재설치하여 nvidia-docker2를 설치했습니다.

 

2. Docker 실행

nvidia/cuda:10.1-cudnn7-devel-ubuntu18.04 이미지를 Pull 후 run 수행

 

 

nvidia-docker run -it --rm nvidia/cuda:10.1-cudnn7-devel-ubuntu18.04 nvidia-smi

해당 이미지는 nvidia 드라이버 418, cuda 10.1이 필요하군요

하지만 제 PC의 nvidia 드라이버는 390... 입니다.. 업그레이드  진행해야 하군요..

 

3. nvidia 그래픽 드라이버 업그레이드

아래와 같이 업그레이드를 위해서 repository를 추가하고 업데이트를 합니다.

sudo add-apt-repository ppa:graphics-drivers/ppa

sudo apt update

software & updates

418이 보이군요... 선택후 Apply Changes를 클릭해 업데이트를 진행합니다.

재부팅후 아래와 같이 버전을 확인하였습니다.

정상 설치되었군요

다시 nvidia-docker를 이용해서 nvidia-smi를 실행해 봅니다.

정상 동작합니다.

이제 Docker에서도 GPU를 사용할 수 있게 되었습니다.

 

반응형