Docker에서 GPU 사용하기

단순하게 Docker 컨테이너 내에서 GPU를 사용할려고 했을때는 오류가 난다 오류를 해결하고 Docker 컨테이너 내에서 GPU를 사용할려면 기존 Docker가 아닌 nvidia-docker2를 설치해줘야 된다.

설치

저장소 및 GPG 키 설정 

$ 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

nvidia-docker2 설치

$ sudo apt-get update
$ sudo apt-get install -y nvidia-docker2

daemon 수정

sudo vi /etc/docker/daemon.json

 

다음 문구를 추가 시켜준다

 

default-runtime": "nvidia"

 

사용법

Docker Compose

 gpu:
    environment:
    runtime: nvidia 
    restart: always

 

runtime: nvidia를 추가 해줘야된다

특정 GPU만을 사용하고 싶을 때는 다음 블로그에 작성하겠습니다