NVIDIA-AI-IOT / tao_toolkit_recipes

Other
32 stars 15 forks source link

Dose not need to pull the tao docker image from ngc? #5

Open dongfangduoshou123 opened 2 years ago

dongfangduoshou123 commented 2 years ago

I view how to use the tao toolkit to train custom model at https://developer.nvidia.com/tao-toolkit. I have not find the step to pull the tao image container, just see a docker login step. I want to ask a question is : only pip install the nvidia-tao package and dose not need to pull the tao docker image from ngc?

thank you!

Tyler-D commented 2 years ago

You could download the docker image directly here: https://catalog.ngc.nvidia.com/orgs/nvidia/teams/tao/containers/tao-toolkit-pyt/tags (action recognition is in this image) https://catalog.ngc.nvidia.com/orgs/nvidia/teams/tao/containers/tao-toolkit-tf/tags