我在寻找一个可以探索机器学习/人工智能的图书馆时遇到了一些困难。我有一对R9 290x,似乎找不到适合它的库。
首先,我尝试了array-fire,它具有出色的CPU性能,但对于机器学习而言却具有较差的GPU性能,如machine_learning示例文件夹中的基准测试所示。
我查看了rocm和MIOpen,尝试了hip enabled tensorflow,但发现290x世代不支持它。我也发现有人在llvm-amdgpu suppport上进行 tensorflow 处理,但看起来还没有准备好
我调查了haskell的加速,发现了有关amdgpu backend的问题,但它似乎还没有准备好。
也许我没有进行足够广泛的搜索?但是据我所知,几乎所有东西都可以在cuda上运行,并且我现在无法为此购买新的GPU。
最佳答案
在您提出问题时,AMD的rocm驱动程序和计算堆栈不支持Hawaii GPU。
从那时起,已经为这些较旧的GPU添加了支持。
AMD制作了一个tensorflow端口,其安装和功能与CUDA tensorflow(amd's port)相同。但是,它不支持gfx803(Fiji,例如R9 Fury)之前的任何版本。
我有一个R9 290,它可以与AMD repo 中的最新rocm驱动程序一起使用,但不能与AMD tensorflow端口一起使用。这是我得到的错误:2018-08-16 12:10:58.529311: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1532] Ignoring visible gpu device (device: 0, name: Hawaii PRO [Radeon R9 290], pci bus id: 0000:01:00.0) with AMDGPU ISA gfx701. The minimum required AMDGPU ISA is gfx803.
关于machine-learning - 尝试在290X上设置机器学习库,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/47898133/