|
|
@@ -131,7 +131,7 @@ case "$gpu_info" in
|
|
|
if [[ $(bc <<< "$pyv <= 3.10") -eq 1 ]]
|
|
|
then
|
|
|
# Navi users will still use torch 1.13 because 2.0 does not seem to work.
|
|
|
- export TORCH_COMMAND="pip install torch==1.13.1+rocm5.2 torchvision==0.14.1+rocm5.2 --index-url https://download.pytorch.org/whl/rocm5.2"
|
|
|
+ export TORCH_COMMAND="pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm5.6"
|
|
|
else
|
|
|
printf "\e[1m\e[31mERROR: RX 5000 series GPUs must be using at max python 3.10, aborting...\e[0m"
|
|
|
exit 1
|
|
|
@@ -141,9 +141,8 @@ case "$gpu_info" in
|
|
|
*"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.6"
|
|
|
- # Navi 3 needs at least 5.5 which is only on the nightly chain, previous versions are no longer online (torch==2.1.0.dev-20230614+rocm5.5 torchvision==0.16.0.dev-20230614+rocm5.5 torchaudio==2.1.0.dev-20230614+rocm5.5)
|
|
|
- # so switch to nightly rocm5.6 without explicit versions this time
|
|
|
+ export TORCH_COMMAND="pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm5.7"
|
|
|
+
|
|
|
;;
|
|
|
*"Renoir"*) export HSA_OVERRIDE_GFX_VERSION=9.0.0
|
|
|
printf "\n%s\n" "${delimiter}"
|
Hina