Kaggle ノートブックで Nvidia TAO ツールキットをセットアップする方法

導入

アクション認識は、ビデオ監視、スポーツ分析、ジェスチャー認識などのアプリケーションを実現する上で重要な役割を果たします。 NVIDIA の TAO ツールキットで事前トレーニングされたモデルを活用すると、高性能のアクション認識モデルを効率的にトレーニングすることが容易になります。

TAO Toolkit は、docker または NGC CLI を使用してセットアップできます。 Kaggle Notebook で作業するため、Kaggle Notebook 環境は docker をサポートしていないため、NGC CLI を使用します。

注: セキュリティ上の懸念、リソース管理、ワークフローを簡略化するための事前構成された環境のプロビジョニングのため、Kaggle Notebooks は Docker をサポートしていません。

インストール手順:

1.依存関係をインストールします

まず、nvidia-pyindex をインストールします。これは、TAO ツールキットと関連コンポーネントのインストール プロセスを簡素化する、NVIDIA の Python ベース ツールのリポジトリ マネージャーです。

!pip install nvidia-pyindex

2. Nvidia TAO Toolkit と NGC-CLI をインストールします

Nvidia TAO ツールキットには、オブジェクトの検出、分類、セグメンテーション、アクション認識などのさまざまなタスク用の事前トレーニングされたモデルのコレクションが含まれています。

!pip install nvidia-tao

次に、NGC-CLI (NVIDIA GPU Cloud Command Line Interface) をインストールします。これは、NVIDIA の NGC カタログと対話して、事前トレーニングされたモデルを管理します。

!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 ノートブックで Nvidia TAO ツールキットをセットアップする方法の詳細内容です。詳細については、PHP 中国語 Web サイトの他の関連記事を参照してください。