我是拥有 AMD GPU(RX 5700、Navi10)的悲惨生物之一。我想使用最新的 PyTorch 库在本地计算机上进行一些深度学习并停止使用云实例。
我在互联网上看到 AMD 承诺在未来 2-4 个月内支持 Navi10(1-2 年前写的帖子),但是,我不认为他们发布了“官方”支持。
我在本地计算机上安装了 ROCm,它实际上检测到了我的 GPU,一切看起来都很好,这是
rocminfo
输出。
我安装了必要的 PyTorch ROCm 版本,但是当我尝试运行代码时,出现以下错误。
hipErrorNoBinaryForGpu:无法找到所有当前设备的代码对象!
我想这是因为 ROCm 仍然不支持 gfx1010 或者我在这一点上迷失了。
如果有人可以提供一种使 ROCm 工作的方法(最好不要再次为 gfx1010 编译整个包)或者提供像 CUDA 用户一样使用 AMD GPU 的方法,我会很高兴。
设置
HSA_OVERRIDE_GFX_VERSION=10.3.0
环境变量。
例如在终端中输入:
$ HSA_OVERRIDE_GFX_VERSION=10.3.0 python launch.py
我使用 5700xt 运行稳定扩散几个月,它有效。
我发现最新的可用的火炬是torch==1.13.1+rocm5.2.0
当然要覆盖 HSA。