Ubuntu

[Docker] Nvidia-docker 설치하기

파송송 2022. 10. 2. 15:31
728x90
  • Container에서 Host OS의 GPU를 사용할 수 없음
  • 이를 사용하기 위해 Nvidia-docker를 설치하는 것임
  • Docker, Nvidia, GPU가 모두 설치되거나 있는 상태에서 환경 세팅을 할 수 있음 (nvidia-smi를 쳐서 확인해보자)

 

1. 기존의 nvidia-docker가 있다면 삭제하고 다시 깐다

$ docker volume ls -q -f driver=nvidia-docker | xargs -r -I{} -n1 docker ps -q -a -f volume={} | xargs -r docker rm -f
$ sudo apt-get purge -y nvidia-docker

2. 레파지토리에 추가한다

$ 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

3. nvidia-docker를 설치한다

$ docker run --runtime=nvidia --rm nvidia/cuda:9.0-base nvidia-smi

4. docker에서 nvidia를 잘 잡는지 확인한다

$ docker run --runtime=nvidia --rm nvidia/cuda:9.0-base nvidia-smi

 

여기서 nvidia-smi과 같은 화면이 뜬다면 성공!

$ vi /etc/docker/daemon.json

docker를 실행하면 자동으로 nvidia-docker가 잡히게 하면 끝

728x90