https://discuss.pytorch.org/t/how-to-run-torch-with-amd-gpu/157069

PyTorch Forums
I usually run my models on Nvidia GPU and I had no problem with torch detecting it. Now I have this GPU: lspci | grep VGA 75eb:00:00.0 VGA compatible controller: Advanced Micro Devices, Inc. [AMD/ATI] Vega 10 [Radeon Instinct MI25 MxGPU] and I’m trying to understand how to make it visible for torch? import torch torch.cuda.is_available() Fal...
How to run torch with AMD gpu?
Was this page helpful?