From 36ac8b8ccaf1bd6d54d0d5da653d62b70fe374b0 Mon Sep 17 00:00:00 2001 From: yihanzhao Date: Thu, 29 Aug 2024 13:59:12 +1000 Subject: [PATCH] fix torchaudio version for amd platform --- scripts/install_torch_amd.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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