Skip to content

Commit

Permalink
Update to ROCm5.7 and PyTorch
Browse files Browse the repository at this point in the history
The webui.sh installs ROCm5.4.2 as default. The webui run failed with AMD
Radeon Pro W7900 with **Segmentation Fault** at Ubuntu22.04 maybe the ABI
compatibility issue.

ROCm5.7 is the latest version supported by PyTorch (https://pytorch.org/)
at now. I test it with AMD Radeon Pro W7900 by PyTorch+ROCm5.7 with PASS.

Signed-off-by: Alex He <heye_dev@163.com>
  • Loading branch information
alexhegit committed Feb 2, 2024
1 parent cf2772f commit db4632f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion webui.sh
Original file line number Diff line number Diff line change
Expand Up @@ -158,7 +158,7 @@ if ! echo "$gpu_info" | grep -q "NVIDIA";
then
if echo "$gpu_info" | grep -q "AMD" && [[ -z "${TORCH_COMMAND}" ]]
then
export TORCH_COMMAND="pip install torch==2.0.1+rocm5.4.2 torchvision==0.15.2+rocm5.4.2 --index-url https://download.pytorch.org/whl/rocm5.4.2"
export TORCH_COMMAND="pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.7"
fi
fi

Expand Down

0 comments on commit db4632f

Please sign in to comment.