NVIDIA Time Series Prediction Platform (TSPP) is a tool designed to compare easily and experiment with arbitrary combinations of forecasting models, time-series datasets, and other configurations. The NVIDIA-TSPP provides functionality to explore the hyperparameter search space, run accelerated model training using distributed training and Automatic Mixed Precision (AMP), and deploy and run inference on accelerated model formats on the NVIDIA Triton Inference Server. The notebooks cover the following topics:
The quick deploy feature automatically sets up the Vertex AI instance with a configuration, preloads the dependencies, and runs the software from NGC without any need to set up the infrastructure.
Note: 4_MultiGpu_HpSearch.ipynb shows multi-GPU training and hp search and requires at least 2 GPUs.
Note: The output of the notebooks was generated using GA100 GPUs on Vertex AI. For faster training, GA100 GPUs are recommended.
To help you get started, we have created sample Jupyter Notebooks that can be easily deployed on Vertex AI using NGC’s One Click Deploy feature. This feature automatically sets up the Vertex AI instance with an optimal configuration, preloads the dependencies, runs the software from NGC without any need to set up the infrastructure.
Simply click on the button that reads “Deploy to Vertex AI” and follow the instructions.
Note: A customized kernel for the Jupyter Notebook is used as the primary mechanism for deployment. This kernel has been built on the Pytorch Container. For more information on the container itself, please refer to this link for more information: NVIDIA-TSPP github