Torchaudio installation fix (NVIDIA#4330)

* separate installer added

Signed-off-by: Aleksandr Laptev <alaptev@nvidia.com>

* apply suggestions, minor fixes

Signed-off-by: Aleksandr Laptev <alaptev@nvidia.com>

Co-authored-by: Aleksandr Laptev <alaptev@nvidia.com>
Signed-off-by: stevehuang52 <heh@nvidia.com>

Dockerfile

@@ -35,13 +35,10 @@ RUN apt-get update && \
 # uninstall stuff from base container
 RUN pip uninstall -y sacrebleu torchtext
 
-# build torchaudio (change latest release version to match pytorch)
+# build torchaudio
 WORKDIR /tmp/torchaudio_build
-RUN git clone --depth 1 --branch release/0.11 https://github.com/pytorch/audio.git && \
-    cd audio && \
-    git submodule update --init --recursive && \
-    BUILD_SOX=1 python setup.py install && \
-    cd .. && rm -r audio
+COPY scripts/installers /tmp/torchaudio_build/scripts/installers/
+RUN /bin/bash /tmp/torchaudio_build/scripts/installers/install_torchaudio_latest.sh
 
 #install TRT tools: PT quantization support and ONNX graph optimizer
 WORKDIR /tmp/trt_build

scripts/installers/install_torchaudio_latest.sh

@@ -0,0 +1,50 @@
+#!/bin/bash
+
+# Copyright (c) 2022, NVIDIA CORPORATION.  All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# Torch and torchaudio versions must match. Othervise, there will be no CUDA support.
+# See https://github.com/pytorch/audio/blob/f0bc00c980012badea8db011f84a0e9ef33ba6c1/README.md?plain=1#L66
+
+TORCHAUDIO_REPO=https://github.com/pytorch/audio
+# expected LATEST_RELEASE=release/*.**
+LATEST_RELEASE=$(git -c 'versionsort.suffix=-' \
+    ls-remote --exit-code --refs --sort='version:refname' --heads ${TORCHAUDIO_REPO} 'release/*.*' \
+    | tail --lines=1 \
+    | cut --delimiter='/' --fields=3,4)
+# expected TORCHAUDIO_BUILD_VERSION=*.**.*
+TORCHAUDIO_BUILD_VERSION=${LATEST_RELEASE:8:1}${PYTORCH_VERSION:1:5}
+
+echo "Latest torchaudio release: ${LATEST_RELEASE:8:4}"
+echo "Pytorch version: ${PYTORCH_VERSION:0:6}"
+echo "Torchaudio build version: ${TORCHAUDIO_BUILD_VERSION}"
+
+# we need parameterized to run torchaudio tests
+# suppose that we do not have parameterized installed yet
+pip install parameterized
+
+# Build torchaudio and run MFCC test
+git clone --depth 1 --branch ${LATEST_RELEASE} https://github.com/pytorch/audio.git && \
+cd audio && \
+git submodule update --init --recursive && \
+BUILD_SOX=1 BUILD_VERSION=${TORCHAUDIO_BUILD_VERSION} python setup.py install && \
+cd .. && \
+pytest -rs audio/test/torchaudio_unittest/transforms/torchscript_consistency_cpu_test.py -k 'test_MFCC' || \
+(echo "ERROR: Failed to install torchaudio!"; exit 1);
+# RNNT loss is built with CUDA, so checking it will suffice
+# This test will be skipped if CUDA is not available (e.g. when building from docker)
+pytest -rs audio/test/torchaudio_unittest/functional/torchscript_consistency_cuda_test.py -k 'test_rnnt_loss' || \
+echo "WARNING: Failed to install torchaudio with CUDA support!";
+rm -rf audio && \
+echo "Torchaudio installed successfully!"