Resource
Kaldi ASR custom backend for the NVIDIA Triton Inference Server.
Use the NGC CLI to download:
Copied!
- Clone the repository.
git clone https://github.com/NVIDIA/DeepLearningExamples.git
cd DeepLearningExamples/Kaldi/SpeechRecognition
- Build the client and server containers.
scripts/docker/build.sh
- Download and set up the pre-trained model and eval dataset.
scripts/docker/launch_download.sh
The model and dataset are downloaded in the data/ folder.
- Start the server.
scripts/docker/launch_server.sh
Once you see the line Starting Metrics Service at 0.0.0.0:8002, the server is ready to be used. You can then start the client.
Currently, multi-GPU is not supported. By default GPU 0 is used. You can use a specific GPU by using NVIDIA_VISIBLE_DEVICES:
NVIDIA_VISIBLE_DEVICES=<GPUID> scripts/docker/launch_server.sh
- Start the client.
The following command will stream 1000 parallel streams to the server. The -p option prints the inferred TEXT sent back from the server.
scripts/docker/launch_client.sh -p