Docker로 Nvidia제 GPU 대응 컨테이너를 사용할 수 있도록 설정한다【Ubuntu 18.04】
Docker는 버전 19.03부터 네이티브로 "Nvidia GPU로 컨테이너 가속"이 지원되었습니다.이 기능은 NVIDIA Container Toolkit을 사용하여 사용할 수 있습니다.여기에서는 필요한 초기 설정을 수행합니다.
환경
OS: 우분투 18.04 LTS
GPU: GTX1080
순서 XNUMX: 전제 조건 확인
Docker의 「GPU에 의한 컨테이너의 가속」을 사용함에 있어서, 환경을 확인해 둡시다.너무 오래된 환경에서는 GPU 컨테이너를 사용할 수 없습니다.
- GNU / Linux x86_64(kernel version > 3.10)← Ubuntu 18.04이므로 OK
- NVIDIA GPU with Architecture > Fermi(2.1 ) ← GPU는 데스크탑용이라면 GTX600대 이후는 OK
- NVIDIA driver 버전 ≧ 361.93 ← 아래의 순서 XNUMX 참조
자세한 내용은 다음 페이지를 참조하십시오.
순서 XNUMX: Docker 설치
아직 Docker를 넣지 않은 경우 다음 단계에 따라 설치합니다.
절차 XNUMX: Nvidia GPU Driver 설치
GPU 컨테이너(CUDA 컨테이너)를 실행하려면 Nvidia 드라이버가 필요합니다.
Nvidia 드라이버의 버전은 사용하고 싶은 CUDA의 버전에 대응하는 것 이상으로 합니다.
Nvidia 드라이버조차 제대로 설치하면 CUDA toolkit이나 NVIDIA Container Toolkit을 별도로 설치하지 않아도 괜찮습니다.
Nvidia Driver를 설치하는 방법은 다음과 같습니다.
XNUMX단계: Docker에서 GPU 컨테이너를 사용할 수 있도록 합니다.
기기에서 다음을 입력합니다.
공식 참조: https://github.com/NVIDIA/nvidia-docker
# 패키지 리포지토리 추가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 sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit sudo systemctl restart docker
동작 확인해보기
# 최신 공식 CUDA 이미지로 테스트 docker run --gpus all nvidia/cuda:9.0-base nvidia-smi
잘하면 이런 표시가 나옵니다.
이후 GPU 컨테이너를 사용하려면 "docker run"에 "-gpus"옵션을 붙입니다.
docker를 설치했다면 docker run 전에 sudo를 추가하지 않으면 실행되지 않을 수 있습니다.
【Ubuntu】도커를 sudo 명령 없이 실행하기 위한 설정과 주의점
이상, Ubuntu에서 Docker의 GPU 대응 컨테이너 이용에 필요한 초기 설정이었습니다.
참고서적