Bagaimana untuk menyediakan Kit Alat Nvidia TAO pada Buku Nota Kaggle

pengenalan

Pengiktirafan tindakan memainkan peranan penting dalam mendayakan aplikasi seperti pengawasan video, analisis sukan dan pengecaman gerak isyarat. Memanfaatkan model pra-latihan dengan TAO Toolkit NVIDIA memudahkan untuk melatih model pengecaman tindakan berprestasi tinggi dengan cekap.

Tao Toolkit boleh disediakan menggunakan docker atau NGC CLI. Memandangkan kami akan mengusahakan Kaggle Notebook, kami akan menggunakan NGC CLI, kerana persekitaran Kaggle Notebook tidak menyokong docker.

Nota: Buku Nota Kaggle tidak menyokong Docker kerana kebimbangan keselamatan, pengurusan sumber dan penyediaan persekitaran yang diprakonfigurasikan untuk aliran kerja yang dipermudahkan.

Langkah Pemasangan:

1. Pasang kebergantungan

Mula-mula, pasang nvidia-pyindex, pengurus repositori untuk alatan berasaskan Python NVIDIA yang memudahkan proses pemasangan untuk Kit Alat TAO dan komponen yang berkaitan.

!pip install nvidia-pyindex

2. Pasang Nvidia TAO Toolkit dan NGC-CLI

Kit Alat Nvidia TAO mengandungi koleksi model terlatih untuk pelbagai tugas seperti pengesanan objek, pengelasan, pembahagian dan pengecaman tindakan.

!pip install nvidia-tao

Seterusnya, pasang NGC-CLI (Antara Muka Barisan Perintah Awan GPU NVIDIA), yang berinteraksi dengan katalog NGC NVIDIA untuk mengurus model pra-latihan.

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

3. Buat akaun NGC

Daftar akaun pada katalog Nvidia NGC untuk mengakses model kit alat TAO. Setelah didaftarkan, anda boleh membuat pengesahan melalui NGC CLI menggunakan kunci API anda untuk memuat turun model yang diingini.

Pertama, pergi ke https://catalog.ngc.nvidia.com/ dan daftar untuk mendapatkan akaun percuma dari menu yang betul.

Setelah log masuk, pergi ke bahagian Persediaan dari menu lungsur kanan dan klik pada Jana Kunci Peribadi.

4. Konfigurasikan NGC CLI

Sediakan persekitaran anda untuk mengesahkan dengan NGC menggunakan arahan berikut. Pastikan kunci API anda selamat.

!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

Jika anda melihat output di bawah, persediaan anda telah selesai. Hore!!??

Sekarang NGC CLI dikonfigurasikan, anda boleh menyenaraikan model yang tersedia:

!ngc-cli/ngc registry model list

Jika anda ingin memuat turun mana-mana model tertentu, anda boleh melakukannya dengan menjalankan arahan berikut

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

Di sini saya telah memuat turun model ActionRecognitionNet. Model akan dimuat turun dalam format .onnx.

Dengan mengikut langkah di atas, anda telah menyediakan Kit Alat TAO pada Buku Nota Kaggle. Kini anda boleh mula meneroka dunia penglihatan komputer berprestasi tinggi dengan mudah.

Selamat Pengekodan!??

Atas ialah kandungan terperinci Bagaimana untuk menyediakan Kit Alat Nvidia TAO pada Buku Nota Kaggle. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!