[NVIDIA TAO] 画像分類モデル作成 - 環境構築編 [1/5]
はじめに NVIDIA TAO(Train, Adapt, and Optimize(旧名 NVIDIA TLT))とはNVIDIAが提供する様々な事前訓練済みのモデルを使用して転移学習を行うことで、ユーザのニーズに合わせたAIモデルを構築する為のツールです。 事前訓練済みモデルには 物体検出(Object Detection) 画像分類(Image Classification) 意図解釈(NLU) 視線推定 テキスト認識 を含めて他大多数の種類が提供されており、 それぞれのモデルをそのまま使うことも出来ますし、ユースケースに合わせて自由に訓練を行い、モデルをカスタマイズすることが出来ます。 今回はその中でも 画像分類(Image Classification) のモデルを使って、訓練や推論に挑戦してみるのを全5つの記事に分けて書いていこうと思います。 目的 自分のライブで撮影された大量の写真をパート毎に分類したい。 AIを使って自動で分類したい。 ローカルで完結したい。(大量の画像が存在して、クラウドにアップロードなんてしてられない) NVIDIAが好きなのでNVIDIA TAO(Train, Adapt, and Optimize)で画像分類モデル(Image Classification)を使ってみたい。 進捗 環境構築 ← 現在地 事前訓練モデルをダウンロード 訓練データの作成 モデルの訓練 推論 ハードウェア要件 ハードウェア 推奨 メモリ 32GB CPU 8コア GPU NVIDIA A100, NVIDIA V100, NVIDIA RTX 30X0 ストレージ 100GB ソフトウェア要件 ソフトウェア バージョン Ubuntu 18.04 LTS docker-ce >19.03.5 docker-API 1.40 nvidia-driver >455 筆者の環境 NVIDIA TAOはUbuntu +...