End-to-end neural acoustic model for automatic speech recognition providing high accuracy at a low memory footprint.
The following section lists the requirements that you need to meet in order to start training the QuartzNet model.
Requirements
This repository contains Dockerfile which extends the PyTorch 21.07-py3 NGC container and encapsulates some dependencies. Aside from these dependencies, ensure you have the following components:
- NVIDIA Docker
- PyTorch 21.07-py3 NGC container
- Supported GPUs:
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 PyTorch
Further required Python packages are listed in requirements.txt, which are automatically installed with the built Docker container. To manually install them, run:
pip install -r requirements.txt
For those unable to use the PyTorch 21.07-py3 NGC container, to set up the required environment or create your own container, see the versioned NVIDIA Container Support Matrix.