有什么方法可以将Tensorflow数据集连接到plaidML-keras以便通过GPU进行深度学习吗?

问题描述

我有一个AMD GPU,并且在运行Windows 10的Python 3.8上运行。 Tensorflow没有在GPU加速方面为AMD用户提供任何支持。 我在StackOverflow上找到了一种描述如何使用plaid-ml在AMD GPU上运行Keras的方法,该方法可以在我的机器上很好地工作(下面的链接),但是并没有成功,因为Keras仅将numpy数组作为模型的输入,而不是张量,尤其是tf.data.Dataset作为输入。

Anyway to work with Keras in Mac with AMD GPU?

我们如何将tf.data.Dataset用作带有plaid-ml后端的Keras模型的输入? 还是有什么方法可以构建Tensorflow 2.x,使其在Windows 10上使用AMD GPU(然后无需将Keras作为独立的)?我见过有人向AMD用户派生Tensorflow存储库,但该版本始终是Tensorflow 1.x。

有人克服了这个问题吗?在CPU上进行深度学习很痛苦!

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)