The NVIDIA container image for Kaldi, release 21.03 is available on NGC.
Contents of the Kaldi container
This container image contains the complete source of the version of Kaldi in /opt/kaldi.
The container also includes the following:
- Ubuntu 20.04 including Python 3.8
- NVIDIA CUDA 11.2.1 including cuBLAS 11.4.1.1026
- NVIDIA cuDNN 8.1.1
- NVIDIA NCCL 2.8.4 (optimized for NVLink™)
- MLNX_OFED 5.1
- OpenMPI 4.0.5
- Nsight Compute 2020.3.0.18
- Nsight Systems 2020.4.3.7
- TensorRT 7.2.2.3
Driver Requirements
Release 21.03 is based on NVIDIA CUDA 11.2.1, which requires NVIDIA Driver release 460.32.03 or later. However, if you are running on Data Center GPUs (formerly Tesla), for example, T4, you may use NVIDIA driver release 418.40 (or later R418), 440.33 (or later R440), 450.51(or later R450). The CUDA driver's compatibility package only supports particular drivers. For a complete list of supported drivers, see the CUDA Application Compatibility topic. For more information, see CUDA Compatibility and Upgrades and NVIDIA CUDA and Drivers Support.
GPU Requirements
Release 21.03 supports CUDA compute capability 6.0 and higher. This corresponds to GPUs in the Pascal, Volta, Turing, and Ampere Architecture GPU families. Specifically, for a list of GPUs that this compute capability corresponds to, see CUDA GPUs. For additional support details, see Deep Learning Frameworks Support Matrix.
Key Features and Enhancements
This Kaldi release includes the following key features and enhancements.
- Kaldi container image version 21.03 is based on Kaldi 376281ca.
- The latest version of NVIDIA CUDA 11.2.1 including cuBLAS 11.4.1.1026
- The latest version of NVIDIA cuDNN 8.1.1
- Ubuntu 20.04 with February 2021 updates.
Packaged scripts
The Kaldi container comes with the following scripts:
-
/workspace/nvidia-examples/librispeech/prepare_data.shwhich downloads a trained model and data. -
/workspace/nvidia-examples/librispeech/run_benchmark.shwhich runs inference on the trained model and data. Setting the environment variable ONLINE=1 will make the benchmark script run in online mode.
NVIDIA Kaldi Container Versions
The following table shows what versions of Ubuntu, CUDA, Kaldi, and TensorRT are supported in each of the NVIDIA containers for Kaldi. For older container versions, refer to the Frameworks Support Matrix.| Container Version | Ubuntu | CUDA Toolkit | Kaldi | TensorRT |
|---|---|---|---|---|
| 21.03 | 20.04 | NVIDIA CUDA 11.2.1 | 5.5 up to commit 376281ca | TensorRT 7.2.2.3 |
| 21.02 | NVIDIA CUDA 11.2.0 | 5.5 up to commit b78fdcaf | TensorRT 7.2.2.3+cuda11.1.0.024 | |
| 20.12 | NVIDIA CUDA 11.1.1 | 5.5 up to commit 7905fe5a | TensorRT 7.2.2 | |
| 20.11 | 18.04 |
NVIDIA CUDA 11.1.0 | 5.5 up to commit edf6963c | TensorRT 7.2.1 |
| 20.10 | 5.5 up to commit 49dab837 | |||
| 20.09 | NVIDIA CUDA 11.0.3 | 5.5 up to commit 8372cc6a | TensorRT 7.1.3 | |
| 20.08 | 5.5 up to commit f88d5a36 | |||
| 20.07 | NVIDIA CUDA 11.0.194 | 5.5 up to commit 67db30cc | ||
| 20.06 | NVIDIA CUDA 11.0.167 | 5.5 up to commit 2c7e78f | TensorRT 7.1.2 | |
| 20.03 | NVIDIA CUDA 10.2.89 | 5.5 up to commit da71f301 | TensorRT 7.0.0 | |
| 5.5 | ||||
| TensorRT 6.0.1 | ||||
| 19.10 | NVIDIA CUDA 10.1.243 | |||
| 19.09 | ||||
| 19.08 | TensorRT 5.1.5 |
Known Issues
There are no known issues in this release.