DeepStream

#2 DeepStream 다짜고짜 실행해보기 - 1

후로링 2020. 11. 9. 17:06

  일단 실행해서 비디오를 분석하는것을 보고 각종 코드, 플러그인에 대해 설명해 보겠습니다. 

 

  DeepStream에 들어가는 각종 프레임워크를 코드단계에서 빌드하는것은 쉬운 일이 아닙니다. 따라서 Container를 활용 하도록 하겠습니다. 제 호스트환경은 Ubuntu 16.04이며 Nvidia-docker가 깔리는 환경이라면 Ubuntu 18.04, CentOS등 어떤 OS를 사용하셔도 상관 없습니다. 

 

1. NGC

  - NGC Container registry : ngc.nvidia.com/catalog/containers/nvidia:deepstream

 

NVIDIA NGC

 

ngc.nvidia.com

  Nvidia에서 개발한 각종 프레임워크, SDK등의 컨테이너를 NGC에서 다운 받을 수 있습니다. 만약 회원가입이 되어있지 않으시다면 회원가입을 하고 로그인을 합니다. 

 

위 메뉴에서 containers를 선택하고, 검색어로 Deepstream을 입력한 후에 Deepstream container로 들어갑니다. 

 

여기서 그냥 docker pull을 하시면 이미지를 찾을 수 없다고 나옵니다. docker 상에서 NGC로그인을 수행해야 하기 때문입니다.

 

2. 컨테이너 설치

Nvidia Docker 설치 : docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html#docker

 

Installation Guide — NVIDIA Cloud Native Technologies documentation

On RHEL 7, install the nvidia-container-toolkit package (and dependencies) after updating the package listing: Restart the Docker daemon to complete the installation after setting the default runtime: Note Depending on how your RHEL 7 system is configured

docs.nvidia.com

Nvidia docker는 Nvidia GPU가 설치되어 있는 환경에서 설치하면 Container내부에서 GPU를 사용할 수 있게 해주는 Docker의 확장입니다. 위 링크에 기본적인 Docker의 설치부터 Nvidia Docker의 설치까지 나와있으며, 그대로 복사/붙여넣기 하신다면 쉽게 설치하실 수 있습니다. 

 

Nvidia Docker를 설치하기 전, 호스트 머신의 GPU Driver version이 440이상인지 먼저 확인해 주세요

nvidia-smi 명령어로 Driver Version을 확인해 봤을때 버전이 낮다면, Nvidia홈페이지에서 드라이버를 업그레이드 하셔야 합니다. 

 

Nvidia Driver와 Nvidia Docker가 모두 설치되어 있다면 이제 NGC에서 계정을 받고 로그인을 하시면 됩니다. 

 

3. NGC 로그인 및 컨테이너 다운로드

오른쪽 상단 메뉴에서 Setup을 클릭하고, Get API Key로 들어갑니다. 보통 Docker hub에 로그인해 컨테이너를 받으실때 그냥 docker login을 사용하셨을텐데, API Key를 받으면 다른 Registry에 로그인 할 수 있습니다. 

 

Generate API Key를 통해 API키를 받으시고, 아래 터미널명령의 빈 곳에 생성한 Key를 넣어주시면 NGC에 로그인 가능합니다. 

 

이후에 위 컨테이너를 받으시면 컨테이너를 받는 절차가 모두 끝나게 됩니다.