I saw this post recommending to use HSA_OVERRIDE_GFX_VERSION=10.3.0 to run Pytorch with ROCm on a 5700XT card, but I could'nt get it to work.
My steps:
sudo pacman -S python-pytorch-opt-rocm
git clone https://github.com/pytorch/examples.git
cd examples/mnist
HSA_OVERRIDE_GFX_VERSION=10.3.0 python3 main.py
Result: I get this output when trying to train mnist. The GPU just runs hot and no training progress is shown.
Note: Also tried with python-pytorch-rocm package, but python-pytorch-opt-rocm should be fine as lscpu shows avx2 support.
So the question is: does anyone know whether this workaround for Navi10 GPUs still works? Or did I miss anything to set this up correctly?
