<aside> 💡 Tao를 사용하기 위해선 Docker와 Nvidia-docker가 설치되어 있어야 함

</aside>

NGC 계정 생성 및 인증

NGC(Nvidia GPU Cloud) NVIDIA에서 제공하는 GPU 소프트웨어 패키지 플랫폼이다.

Tao Toolkit은 NGC에서 제공하는 GPU Optimized Image들을 사용하는데, 이 Image들이 저장된 Docker hub에 로그인 하기 위해선 NGC 계정이 필요하다.

가입 완료 후 Setup 클릭하여 나만의 API Key를 만든다.

Docker Login

위에서 얻은 API Key로 NGC 허브에 로그인한다.

docker login [nvcr.io](<http://nvcr.io/>)

가상 환경 안에서 노트북 설치

virtualenv나 conda 등을 이용하여 가상환경을 만들고 노트북을 설치한다. 노트북을 설치하는 이유는 Tao가 제공하는 모든 Example들을 다운 받을 수 있기 때문이다. 최신 버전은 공식 사이트를 참고하길 바란다.

wget --content-disposition <https://api.ngc.nvidia.com/v2/resources/nvidia/tao/cv_samples/versions/v1.4.0/zip> -O cv_samples_v1.4.0.zip
unzip -u cv_samples_v1.4.0.zip  -d ./cv_samples_v1.4.0 && rm -rf cv_samples_v1.4.0.zip && cd ./cv_samples_v1.4.0

완료되면 cv_samples_v1.4.0 폴더 안에 수많은 Example들이 있는 것을 확인할 수 있다.

예제들을 쉽게 보기 위해 노트북을 설치한다. Jupyter notebook은 vs code에서 Plug-in을 설치하면 호환이 잘 되니 굳이 별도로 띄워놓고 사용하지 않아도 된다.

pip3 install jupyter
jupyter notebook --ip 0.0.0.0 --allow-root --port 8888

[VS Code Plug-in]