본문 바로가기

Programing/Linux

[완료] Nvidia Docker (GPU 사용) 설치



그냥 일반 Docker 를 사용할 경우 GPU 사용에 제한이 있다.

따라서 그래픽요소가 많이 사용되는 프로그램을 쓰거나, 딥러닝과 같은 연산이 필요한 도커를 실행시킬 경우

GPU를 이어주는 Nvidia-Docker 툴을 설치&사용해야 한다.


신규버전이나 사용법은 Nvidia-Docker Github에서 확인하면 된다.

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


[ Index ]

#1. 기존 설치내역 삭제(nvidia-docker 버전1 삭제)

#2. Nvidia Docker 패키지 repository 등록

#3. Nvidia-Docker 2 버전 설치 & 데몬 설정

#4. Nvidia Docker 실행 테스트

#5. Nvidia 연동내역 확인





#1. 기존 설치내역 삭제(nvidia-docker 버전1 삭제)

     > nvidia-docker 드라이버 사용 볼륨 제거, 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

  *xargs : 앞의 명령어 출력결과를 다음 명령어의 인자(argument)로 사용



#2. Nvidia Docker 패키지 repository 등록

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


#3. Nvidia-Docker 2 버전 설치 & 데몬 설정

sudo apt-get install -y nvidia-docker2
sudo pkill -SIGHUP dockerd


#4. Nvidia Docker 실행 테스트

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


#5. Nvidia 연동내역 확인

nvidia

docker-ce > nvidia-docker


https://hiseon.me/linux/ubuntu/install-docker/

https://cnvrg.io/how-to-setup-docker-and-nvidia-docker-2-0-on-ubuntu-18-04/