Kaldi Release 20.01
The NVIDIA container image for Kaldi, release 20.01, 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 18.04 including Python 3.6
- NVIDIA CUDA 10.2.89 including cuBLAS 10.2.2.89
- NVIDIA cuDNN 7.6.5
- NVIDIA NCCL 2.5.6 (optimized for NVLink™ )
- MLNX_OFED
- OpenMPI 3.1.4
- Nsight Compute 2019.5.0
- Nsight Systems 2019.6.1
- TensorRT 7.0.0
Driver Requirements
Release 20.01 is based on NVIDIA CUDA 10.2.89, which requires NVIDIA Driver release 440.33.01. However, if you are running on Tesla (for example, T4 or any other Tesla board), you may use NVIDIA driver release 396, 384.111+, 410, 418.xx or 440.30. 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.
GPU Requirements
Release 20.01 supports CUDA compute capability 6.0 and higher. This corresponds to GPUs in the Pascal, Volta, and Turing 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 20.01 is based on Kaldi 5.5.
- Commit hash: ba92f608
- This contains minor bug and performance fixes.
- Latest version of TensorRT 7.0.0
- Ubuntu 18.04 with December 2019 updates.
Packaged scripts
The Kaldi container comes with the following scripts:
-
/workspace/nvidia-examples/librispeech/prepare.sh
which downloads a trained model and data. -
/workspace/nvidia-examples/librispeech/run_benchmark.sh
and/workspace/nvidia-examples/librispeech/run_multigpu_benchmark.sh
which run inference on the trained model and data.
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 |
---|---|---|---|---|
20.01 | 18.04 16.04 |
NVIDIA CUDA 10.2.89 | 5.5 | TensorRT 7.0.0 |
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.