试图在290X上设置机器学习库

时间:2017-12-20 02:54:48

标签: machine-learning tensorflow arrayfire amd-gpu

我在寻找用于探索机器学习/ ai的库时遇到了一些困难。我有一对R9 290x,似乎找不到适合它的lib。

首先,我尝试使用array-fire,它具有出色的CPU性能,但机器学习的GPU性能较差,如machine_learning示例文件夹中的基准测试所示。

我查看了rocm和MIOpen,我尝试了hip enabled tensorflow,但发现290x代不支持它。我发现有人在llvm-amdgpu suppport处理张力流,但它还没有准备好

我调查了针对haskell的加速,发现了amdgpu backend的问题,但它看起来还没有准备好。

也许我的搜索范围不够广泛?但据我所知,几乎所有东西都在cuda上运行,我现在买不起新的GPU。

1 个答案:

答案 0 :(得分:1)

在您提出问题时,AMD的rocm驱动程序和计算堆栈不支持Hawaii GPU。 从那时起,已经为这些较旧的GPU添加了支持。

AMD制作了一个tensorflow端口,该端口的安装和功能与CUDA tensorflow(amd's port)相同。但是,它不支持gfx803(Fiji,例如R9 Fury)之前的任何版本。

我有一个R9 290,它可以与AMD回购中的最新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.