Kaggle Notebook에서 Nvidia TAO 툴킷을 설정하는 방법

소개

동작 인식은 비디오 감시, 스포츠 분석, 동작 인식과 같은 애플리케이션을 구현하는 데 중요한 역할을 합니다. NVIDIA의 TAO 툴킷으로 사전 훈련된 모델을 활용하면 고성능 동작 인식 모델을 더 쉽고 효율적으로 훈련할 수 있습니다.

TAO 툴킷은 docker 또는 NGC CLI를 사용하여 설정할 수 있습니다. Kaggle Notebook을 작업할 예정이므로 Kaggle Notebook 환경은 docker를 지원하지 않으므로 NGC CLI를 사용하겠습니다.

참고: Kaggle Notebooks는 보안 문제, 리소스 관리 및 단순화된 워크플로를 위한 사전 구성된 환경 제공으로 인해 Docker를 지원하지 않습니다.

설치 단계:

1. 종속성 설치

먼저 TAO 툴킷 및 관련 구성 요소의 설치 프로세스를 단순화하는 NVIDIA의 Python 기반 도구용 저장소 관리자인 nvidia-pyindex를 설치합니다.

!pip install nvidia-pyindex

2. Nvidia TAO 툴킷 및 NGC-CLI 설치

Nvidia TAO 툴킷에는 객체 감지, 분류, 분할, 동작 인식 등 다양한 작업을 위해 사전 훈련된 모델 모음이 포함되어 있습니다.

!pip install nvidia-tao

다음으로 NVIDIA의 NGC 카탈로그와 상호작용하여 사전 훈련된 모델을 관리하는 NGC-CLI(NVIDIA GPU 클라우드 명령줄 인터페이스)를 설치하세요.

!wget -O ngccli_linux.zip https://ngc.nvidia.com/downloads/ngccli_linux.zip
!unzip ngccli_linux.zip

3. NGC 계정 만들기

TAO 툴킷 모델에 액세스하려면 Nvidia NGC 카탈로그에 계정을 등록하세요. 등록한 후에는 API 키를 사용하여 NGC CLI를 통해 인증하여 원하는 모델을 다운로드할 수 있습니다.

먼저 https://catalog.ngc.nvidia.com/에 접속해 오른쪽 메뉴에서 무료 계정을 가입하세요.

로그인한 후 오른쪽 드롭다운 메뉴에서 설정 섹션으로 이동하여 개인 키 생성을 클릭하세요.

4. NGC CLI 구성

다음 명령을 사용하여 NGC를 인증할 수 있는 환경을 설정하세요. API 키를 안전하게 보관하세요.

!chmod u+x ngc-cli/ngc

import os

# Declaring the input arguments as environment variables. 
# This way we can directly pass the arguments during cell runtime to any command request in the Kaggle notebook.

os.environ['API_KEY'] = 'your_api_key'
os.environ['TYPE'] = 'ascii'
os.environ['ORG'] = '0514167173176982'
os.environ['TEAM'] = 'no-team'
os.environ['ACE'] = 'no-ace'

# Passing the input arguments to the config command
!echo -e "$API_KEY\n$TYPE\n$ORG\n$TEAM\n$ACE" | ngc-cli/ngc config set

아래 출력이 보이면 설정이 완료된 것입니다. 만세!!??

이제 NGC CLI가 구성되었으므로 사용 가능한 모델을 나열할 수 있습니다.

!ngc-cli/ngc registry model list

특정 모델을 다운로드하려면 다음 명령을 실행하면 됩니다

!ngc-cli/ngc registry model download-version "nvidia/tao/actionrecognitionnet:deployable_onnx_v2.0"

여기에서 ActionRecognitionNet 모델을 다운로드했습니다. 모델은 .onnx 형식으로 다운로드됩니다.

위 단계에 따라 Kaggle Notebook에 TAO 툴킷을 설정했습니다. 이제 고성능 컴퓨터 비전의 세계를 쉽게 탐색할 수 있습니다.

즐거운 코딩 되세요!??

위 내용은 Kaggle Notebook에서 Nvidia TAO 툴킷을 설정하는 방법의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!