This project contains the source code to the TAO launcher interface. This launcher interface aims at providing a unified command line experience to the Transfer Learning Toolkit package. The DNN's in TAO may be implemented in TensorFlow, Keras or PyTorch. These frameworks are difficult to maintain in the same docker. In an attempt to abstract the final customers of TAO, to handle tao command abstracts these details away from the user.
This section covers a quick guide to start working with the launcher.
| Software | Version |
|---|---|
| python | >= 3.6.9 |
| docker-ce | >19.03.5 |
| docker-API | 1.40 |
| nvidia-container-toolkit | >1.3.0-1 |
| nvidia-container-runtime | 3.4.0-1 |
| nvidia-docker2 | 2.5.0-1 |
| nvidia-driver | >450 |
-
Install docker-ce by following the official instructions.
Once you have installed docker-ce, please follow the post-installation steps to make sure that the docker can be run without sudo.
-
Install nvidia-container-toolkit by following these installation instructions.
-
Login to the NGC staging area docker registry using the api key generated from
ngc.nvidia.com
docker login nvcr.io- The TAO Launcher can be installed using the wheel from the google drive with the command mentioned below. We highly recommend using python virtualenv to isolate your python env.
pip install nvidia_tao.0.0.9.dev0-py3x-none-any.whlWhere x is the minor version of the python installed in your PC.
- Initialize the launcher using
tao init.
2020-11-25 14:07:54,322 [WARNING] __main__:
************************************************************************************************************************************************************
A config file wasn't found to initialize a TAO instance. Please run the tao init command to get the config file and configure the DL tasks available in TAO.
************************************************************************************************************************************************************
2020-11-25 14:07:54,325 [INFO] tao.components.instance_handler.utils: Downloading the latest config file.
2020-11-25 14:07:54,459 [INFO] tao.components.instance_handler.utils: Download complete. Validating the file.
- Run the tao tasks using the changed tao cli structure.
tao <task> <sub-task> <cli-args>For detailed instructions on using the launcher, please navigate to this page.
This project is licensed under the Apache 2.0 License.