안녕하세요.
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 설치를 위한 소스 리스트를 추가합니다.
본문에는 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 |
하지만 제 PC의 nvidia 드라이버는 390... 입니다.. 업그레이드 진행해야 하군요..
3. nvidia 그래픽 드라이버 업그레이드
아래와 같이 업그레이드를 위해서 repository를 추가하고 업데이트를 합니다.
sudo add-apt-repository ppa:graphics-drivers/ppa sudo apt update |
418이 보이군요... 선택후 Apply Changes를 클릭해 업데이트를 진행합니다.
재부팅후 아래와 같이 버전을 확인하였습니다.
정상 설치되었군요
다시 nvidia-docker를 이용해서 nvidia-smi를 실행해 봅니다.
정상 동작합니다.
이제 Docker에서도 GPU를 사용할 수 있게 되었습니다.
'Dev Platform > Docker' 카테고리의 다른 글
[Docker] tensorflow object detection 샘플 돌려보기(GPU) (1) | 2019.04.07 |
---|---|
[Docker] jupyter notebook(쥬피터 노트북) 포함하여 이미지 만들기 (0) | 2019.04.03 |
[Docker] python3 도커 이미지 만들기 ( 파이참 포함 ) (1) | 2019.03.31 |
[Docker] image를 계층 분리해서 관리 ( Dockerfile ) (0) | 2019.03.31 |
[Docker] Docker를 이용해서 QT 빌드 시스템 구축해보기 (0) | 2019.03.27 |