The following section list the requirements in order to start training the SSD320 v1.2 model.

Requirements

This repository contains Dockerfile which extends the TensorFlow NGC container and encapsulates some dependencies. Aside from these dependencies, ensure you have the following software:

• NVIDIA Docker
• TensorFlow 20.06-py3 (or later) NGC container
• GPU-based architecture:
• NVIDIA Volta
• NVIDIA Turing
• NVIDIA Ampere architecture

For more information about how to get started with NGC containers, see the following sections from the NVIDIA GPU Cloud Documentation and the Deep Learning Documentation:

• Getting Started Using NVIDIA GPU Cloud
• Accessing And Pulling From The NGC Container Registry
• Running TensorFlow