Add torchaudio wheel (#3602)

Author: sramakintel

docker/Dockerfile.prebuilt

@@ -94,18 +94,21 @@ RUN ln -sf $(which ${PYTHON}) /usr/local/bin/python && \
 
 ARG TORCH_VERSION
 ARG TORCHVISION_VERSION
+ARG TORCHAUDIO_VERSION
 ARG IPEX_VERSION
 ARG ONECCL_BIND_PT_VERSION
 ARG TORCH_WHL_URL
 ARG IPEX_WHL_URL
 ARG TORCHVISION_WHL_URL
+ARG TORCHAUDIO_WHL_URL
 ARG ONECCL_BIND_PT_WHL_URL
 
 RUN python -m pip install numpy
 
 RUN python -m pip install torch==${TORCH_VERSION} --extra-index-url ${TORCH_WHL_URL} && \
     python -m pip install intel_extension_for_pytorch==${IPEX_VERSION} --extra-index-url ${IPEX_WHL_URL} && \
     python -m pip install torchvision==${TORCHVISION_VERSION} --extra-index-url ${TORCHVISION_WHL_URL} && \
+    python -m pip install torchaudio==${TORCHAUDIO_VERSION} --extra-index-url ${TORCHAUDIO_WHL_URL} && \
     python -m pip install oneccl_bind_pt==${ONECCL_BIND_PT_VERSION} --extra-index-url ${ONECCL_BIND_PT_WHL_URL} 
 
 ENV LD_LIBRARY_PATH=/opt/intel/oneapi/lib:/opt/intel/oneapi/lib/intel64:$LD_LIBRARY_PATH

docker/build.sh

@@ -18,9 +18,11 @@ if [[ ${IMAGE_NAME} != "" ]]; then
                  --build-arg TORCH_VERSION=2.1.0a0+cxx11.abi \
                  --build-arg IPEX_VERSION=2.1.10+xpu \
                  --build-arg TORCHVISION_VERSION=0.16.0a0+cxx11.abi \
+                 --build-arg TORCHAUDIO_VERSION=2.1.0a0+cxx11.abi \
                  --build-arg TORCH_WHL_URL=https://pytorch-extension.intel.com/release-whl/stable/xpu/us/ \
                  --build-arg IPEX_WHL_URL=https://pytorch-extension.intel.com/release-whl/stable/xpu/us/ \
                  --build-arg TORCHVISION_WHL_URL=https://pytorch-extension.intel.com/release-whl/stable/xpu/us/ \
+                 --build-arg TORCHAUDIO_WHL_URL=https://pytorch-extension.intel.com/release-whl/stable/xpu/us/ \
                  --build-arg CCL_VER=2021.11.1-6 \
                  --build-arg ONECCL_BIND_PT_VERSION=2.1.100 \
                  --build-arg ONECCL_BIND_PT_WHL_URL=https://pytorch-extension.intel.com/release-whl/stable/xpu/us/ \