amd gpu + pytorch 选哪个 device？

Question

from torch import device

device('cpu')

pytorch 支持很多 device：
cpu, cuda, ipu, xpu, mkldnn, opengl, opencl, ideep, hip, ve, fpga, ort, xla, lazy, vulkan, mps, meta, hpu

---

nvidia 的选 cuda
Apple Silicon 选 mps

amd 的 gpu 选什么？难道也是 cuda？

cuda 不是 nvidia 独享的吗？

amd gpu 做了 cuda 兼容？

或者说 nvidia 把 cuda 开放给 amd gpu 了？

Answer 1

可以试试 opengl