pin version for torch for Navi3 according to comment from #11228

    *"Navi 2"*) export HSA_OVERRIDE_GFX_VERSION=10.3.0

    ;;

    *"Navi 3"*) [[ -z "${TORCH_COMMAND}" ]] && \

        export TORCH_COMMAND="pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm5.5"

        export TORCH_COMMAND="pip install --pre torch==2.1.0.dev-20230614+rocm5.5 torchvision==0.16.0.dev-20230614+rocm5.5 --index-url https://download.pytorch.org/whl/nightly/rocm5.5"

        # Navi 3 needs at least 5.5 which is only on the nightly chain

    ;;

    *"Renoir"*) export HSA_OVERRIDE_GFX_VERSION=9.0.0