[Vitis-AI] 개발 키트 다운로드 - 1 (Docker 설치 + 호스트 설정)

Vitis-AI 를 사용해보게 되어서 사용법도 익히고 공부할 수 있는 환경을 구성해보려고 하였습니다.

 

Xilinx의 Vitis AI User Documentation 를 통해서 진행 해보았습니다.

 

 

www.xilinx.com/html_docs/vitis_ai/1_3/installation.html#qoi1603967292432

Installation and Setup

 

---

      1. Xilinx gitbub를 통해서 Vitis-AI 패키지 다운로드

 

github.com/Xilinx/Vitis-AI

Xilinx/Vitis-AI

 

   2. Docker 설치

 

• Docker Repository 추가

 

sudo add-apt-repository "deb [arch=amd64] https://download.docker.com/linux/ubuntu bionic stable" 

curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -

sudo apt-get update && sudo apt install docker-ce docker-ce-cli containerd.io

 

•  Nvidia Docker Rntime 설치

curl -s -L https://nvidia.github.io/nvidia-container-runtime/gpgkey | \     
   sudo apt-key add -

curl -s -L https://nvidia.github.io/nvidia-container-runtime/$distribution/nvidia-container-runtime.list | \
   sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list

sudo apt-get update

sudo apt-get install nvidia-container-runtime

 

• Docker Config 파일 수정

sudo systemctl edit docker

 

- 아래의 코드 추가 후 exit

[Service] 
ExecStart= 
ExecStart=/usr/bin/dockerd --group docker -H unix:///var/run/docker.sock --add-runtime=nvidia=/usr/bin/nvidia-container-runtime

 

• Daemon과  Docker 재실행

sudo systemctl daemon-reload

sudo systemctl restart docker

 

 

 

• Group Docker 설정  / 원문 :docs.docker.com/engine/install/linux-postinstall/

sudo groupadd docker

sudo usermod -aG docker $USER

newgrp docker 

docker run hello-world

 

 

3.  호스트 설정

•  Docker 컨테이너 실행 

• (Docker Hub에서 CPU 이미지 실행)

docker pull xilinx/vitis-ai:latest

./docker_run.sh xilinx/vitis-ai

 

 

 

다운로드가 되나봅니다... 그리고 다시한번 입력하면 

 

 

 

이런 화면이 나오게 되네요

 

 

저는 ultrascale ultra96 보드를 사용 예정 이기에

 

Edge 용으로 설정하게 됩니다.

 

 

 

www.xilinx.com/bin/public/openDownload?filename=sdk-2020.2.0.0.sh

 

위 링크를 클릭하여  "sdk-2020.2.0.0.sh" 를 다운로드 합니다.

 

다운 받은 것을 Vitis-AI-master 폴더에 넣습니다. 그 다음

 

 

 

 - 크로스 컴파일 환경 설치

./sdk-2020.2.0.0.sh

 

이 단게에서 파일에 대한 권한 설정을 필요로 할 수도 있습니다. 

 

 

- 파일에 대한 권한 부여

chmod 755 sdk-2020.2.0.0.sh

이제 다시 한 번  - 크로스 컴파일 환경 설치

./sdk-2020.2.0.0.sh

 

 

~petalinux_sdk 디렉토리 설정 해주는데 "~" 빼지 않을 것...

 

 

 

- 설치가 완료되면 다음 명령을 입력

source ~/petalinux_sdk/environment-setup-aarch64-xilinx-linux

 

 

 

www.xilinx.com/bin/public/openDownload?filename=vitis_ai_2020.2-r1.3.0.tar.gz

 

위 링크를 통해서 vitis_ai_2020.2-r1.3.0.tar.gz 를 다운로드하여 이것 또한 Vitis-AI-master 폴더에 저장

 

 

- 설치 완료 후 명령 입력

tar -xzvf vitis_ai_2020.2-r1.3.0.tar.gz -C ~/petalinux_sdk/sysroots/aarch64-xilinx-linux

 

- resnet50 샘플을 교차 컴파일하여 확인

cd Vitis-AI/demo/VART/resnet50

bash –x build.sh

 

 

 

 

 

위와 같은 화면이 나오면 여기 까지 호스트 설정 까진 완료..