remove not necessay explicit version for torch torchvision and remove torchaudio

This commit is contained in:
Christian Bayle 2025-01-10 13:50:03 +01:00
parent a2b91f3741
commit 57722f86e6

View File

@ -153,7 +153,7 @@ case "$gpu_info" in
*"Navi 2"*) export HSA_OVERRIDE_GFX_VERSION=10.3.0
;;
*"Navi 3"*"RX 76"*) [[ -z "${TORCH_COMMAND}" ]] && \
export TORCH_COMMAND="pip install --extra-index-url https://download.pytorch.org/whl/rocm5.7 torch==2.3.1+rocm5.7 torchvision==0.18.1+rocm5.7 torchaudio==2.3.1+rocm5.7" && \
export TORCH_COMMAND="pip install torch torchvision --index-url https://download.pytorch.org/whl/rocm5.7" && \
export HSA_OVERRIDE_GFX_VERSION=11.0.0
;;
*"Navi 3"*) [[ -z "${TORCH_COMMAND}" ]] && \