在AMD APU中运行pytorch或tensorflow

Question

我有一台配备 AMD RADEON APU 的机器：gfx90c 我正在使用 arch linux。

我一直在努力让pytorch和tensorflow使用APU，但到目前为止还没有结果。 我已经尝试构建tensorflow romc的arch包但什么也没有。 有人知道如何实现这一目标吗？

Answer 1

如果您使用的是 ArchLinux： https://bugs.archlinux.org/task/79620

各位，2023 年 9 月，ROCM 不支持 AMD APU。 所以不可能在AMD APU中使用pytorch或tensorflow。