I’m hoping to use PyTorch with ROCm to speed up some SVD using an AMD GPU. I’m new to GPU computing, ROCm and PyTorch, and feel a bit lost.
I’m pretty sure I need ROCm >= 5.0 to support the 6800 RX GPU, which means the PyTorch Get Started Locally command doesn’t quite work for me. Further, I’d like to test on a laptop with a Vega 8 iGPU which some ROCm packages do not support (HID, I believe).
Hence, I need to install ROCm differently, and due to my OS, I can’t use the AMD script (PopOS 21.10), but go through the Using Package Manager on Ubuntu method, which is all fine.
In the very last step, I’m asked to install ROCm meta-packages—and here I’m unsure what I need for PyTorch. AMD has a neat list of ROCm meta-packages, but I don’t know which are necessary.
I’ve been trying to prod the URL from the PyTorch Get Started Locally command, but I didn’t get anything out of that.
So, in short: Which of the ROCm meta-packages should I install to make use of PyTorch?
Thank you for your time.