diff --git a/scripts/install_torch_amd.sh b/scripts/install_torch_amd.sh
index 7696203..df9ef3e 100644
--- a/scripts/install_torch_amd.sh
+++ b/scripts/install_torch_amd.sh
@@ -7,5 +7,5 @@ if ! [[ "$TARGETPLATFORM" ]]; then
 fi
 
 if [[ "$TARGETPLATFORM" != "linux/arm64" ]]; then
-  pip3 install --no-cache-dir torch==1.12.1+cu113 torchvision==0.13.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113 --upgrade
+  pip3 install --no-cache-dir torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113 --upgrade
 fi
\ No newline at end of file