|
|
|
@ -104,6 +104,23 @@ then
|
|
|
|
fi
|
|
|
|
fi
|
|
|
|
|
|
|
|
|
|
|
|
# Check prerequisites
|
|
|
|
# Check prerequisites
|
|
|
|
|
|
|
|
gpu_info=$(lspci 2>/dev/null | grep VGA)
|
|
|
|
|
|
|
|
case "$gpu_info" in
|
|
|
|
|
|
|
|
*"Navi 1"*|*"Navi 2"*) export HSA_OVERRIDE_GFX_VERSION=10.3.0
|
|
|
|
|
|
|
|
;;
|
|
|
|
|
|
|
|
*"Renoir"*) export HSA_OVERRIDE_GFX_VERSION=9.0.0
|
|
|
|
|
|
|
|
printf "\n%s\n" "${delimiter}"
|
|
|
|
|
|
|
|
printf "Experimental support for Renoir: make sure to have at least 4GB of VRAM and 10GB of RAM or enable cpu mode: --use-cpu all --no-half"
|
|
|
|
|
|
|
|
printf "\n%s\n" "${delimiter}"
|
|
|
|
|
|
|
|
;;
|
|
|
|
|
|
|
|
*)
|
|
|
|
|
|
|
|
;;
|
|
|
|
|
|
|
|
esac
|
|
|
|
|
|
|
|
if echo "$gpu_info" | grep -q "AMD" && [[ -z "${TORCH_COMMAND}" ]]
|
|
|
|
|
|
|
|
then
|
|
|
|
|
|
|
|
export TORCH_COMMAND="pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.2"
|
|
|
|
|
|
|
|
fi
|
|
|
|
|
|
|
|
|
|
|
|
for preq in "${GIT}" "${python_cmd}"
|
|
|
|
for preq in "${GIT}" "${python_cmd}"
|
|
|
|
do
|
|
|
|
do
|
|
|
|
if ! hash "${preq}" &>/dev/null
|
|
|
|
if ! hash "${preq}" &>/dev/null
|
|
|
|
@ -164,16 +181,6 @@ then
|
|
|
|
else
|
|
|
|
else
|
|
|
|
printf "\n%s\n" "${delimiter}"
|
|
|
|
printf "\n%s\n" "${delimiter}"
|
|
|
|
printf "Launching launch.py..."
|
|
|
|
printf "Launching launch.py..."
|
|
|
|
printf "\n%s\n" "${delimiter}"
|
|
|
|
printf "\n%s\n" "${delimiter}"
|
|
|
|
gpu_info=$(lspci 2>/dev/null | grep VGA)
|
|
|
|
exec "${python_cmd}" "${LAUNCH_SCRIPT}" "$@"
|
|
|
|
if echo "$gpu_info" | grep -q "AMD"
|
|
|
|
|
|
|
|
then
|
|
|
|
|
|
|
|
if [[ -z "${TORCH_COMMAND}" ]]
|
|
|
|
|
|
|
|
then
|
|
|
|
|
|
|
|
export TORCH_COMMAND="pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.2"
|
|
|
|
|
|
|
|
fi
|
|
|
|
|
|
|
|
HSA_OVERRIDE_GFX_VERSION=10.3.0 exec "${python_cmd}" "${LAUNCH_SCRIPT}" "$@"
|
|
|
|
|
|
|
|
else
|
|
|
|
|
|
|
|
exec "${python_cmd}" "${LAUNCH_SCRIPT}" "$@"
|
|
|
|
|
|
|
|
fi
|
|
|
|
|
|
|
|
fi
|
|
|
|
fi
|
|
|
|
|