Unverified Commit 4599e8ad authored by DaniAndTheWeb's avatar DaniAndTheWeb Committed by GitHub
Browse files

Environment variable on launch just for Navi cards 

Setting HSA_OVERRIDE_GFX_VERSION=10.3.0 for all AMD cards seems to break compatibility for polaris and vega cards so it should just be enabled on Navi
parent d1ea518d

webui.sh

+6 −1
Original line number Diff line number Diff line
@@ -172,8 +172,13 @@ else 
        then	    

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

        fi

        if echo "$gpu_info" | grep -q "Navi"

        then

            HSA_OVERRIDE_GFX_VERSION=10.3.0 exec "${python_cmd}" "${LAUNCH_SCRIPT}" "$@"

        else

            exec "${python_cmd}" "${LAUNCH_SCRIPT}" "$@"

        fi

    else

        exec "${python_cmd}" "${LAUNCH_SCRIPT}" "$@"

    fi

fi

CRA Git | Maintained and supported by SUSTech CRA and CCSE