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

Question

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

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

Answer 1

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

不，根据pypi，最新版本（即 2020 年 9 月 12 日）tensorflow-directml 1.15.3.dev200911可供公众使用。有关更多详细信息，请参阅此。

在 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

有关更多信息，请参阅此。