python - 如何在 directml 后端使用 tensorflow v2

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

Is there another version of "tensorflow-directml" that uses tensorflowv2

不，根据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 .