如何将 tensorflow v2 与 directml 后端一起使用

【问题标题】：How to use tensorflow v2 with directml backend如何将 tensorflow v2 与 directml 后端一起使用
【发布时间】：2021-02-03 18:19:15
【问题描述】：

我有一台带有 amd gpu (rx 5600 xt) 的 windows 操作系统的计算机，我想在 gpu 上运行 tensorflow。我找到了“tensorflow-directml”，它允许我在我的 gpu 上运行 tensorflow，但它使用的是 tensorflow 1.14.0。是否有另一个版本的“tensorflow-directml”使用 tensorflow v2，或者是否有另一种方法可以在我的 gpu 中运行 tensorflow？

谢谢，如果我写错或不准确的地方很抱歉

【问题讨论】：

标签： python tensorflow deep-learning tensorflow2.0 amd-gpu

【解决方案1】：

是否有另一个版本的“tensorflow-directml”使用 tensorflow v2

不，根据pypi，最新版本（即 2020 年 9 月 12 日）tensorflow-directml 1.15.3.dev200911 可供公众使用。更多详情请参考this。

在 Windows 上的 GPU 中运行 Tensorflow

对于 TensorFlow 1.x（即对于 1.15 及更早版本，CPU 和 GPU 包是分开的）

pip install tensorflow-gpu==1.15  # GPU

对于 Tensorflow 2.x（即 V2）及更高版本，pip 包包括对启用 CUDA 的卡的 GPU 支持

pip install tensorflow

更多信息请参考this。

【讨论】：

感谢您的回答，但我有 AMD GPU。我将尝试在 Ubuntu 虚拟机上将它与 Rocm 一起使用。