HSA_OVERRIDE_GFX_VERSION=10.3.0 to force using the Navi21 binary slice.
This is totally unsupported and please don't complain if something doesn't work when using that trick.
But basic PyTorch use works using this, so you might get away with it for this scenario.
(TL;DR: AMD just doesn't care about GPGPU on the mainstream, better to switch to another GPU vendor that does next time...)
HSA_OVERRIDE_GFX_VERSION=10.3.0 to force using the Navi21 binary slice.
This is totally unsupported and please don't complain if something doesn't work when using that trick.
But basic PyTorch use works using this, so you might get away with it for this scenario.
(TL;DR: AMD just doesn't care about GPGPU on the mainstream, better to switch to another GPU vendor that does next time...)