如何在配备 AMD GPU 的 Windows 上使用 TensorFlow？

Question

我想在 Windows (Win 10) 上使用 TensorFlow 和 AMD GPU。

如果我用谷歌搜索，有很多讨论和资源，但我只是不知道目前最好的方法是什么。

有人可以写一个简短的安装说明，他认为这是最好和最新的方法吗？

Answer 1

Tensorflow 官方仅支持 CUDA，这是 NVIDIA 专有技术。有一个使用 openCL here 的非官方实现可以工作，或者您可以尝试使用 Google colab

Answer 2

6年后遇到了完全相同的问题。系统：Win 10 Pro，64 位，AMD Ryzen 5 PRO 5675U

现在有了 Tensorflow2。 这里是安装说明。我使用了 pip 指令并在几分钟内安装了它。我想我应该分享它以节省其他人寻找解决方案的时间。

# Requires the latest pip
pip install --upgrade pip

# Current stable release for CPU and GPU
pip install tensorflow

我事先在 Windows Powershell 中安装了 WSL2

wsl --install

（说明）。 Windows 也有说明。它们可能对其他人有用。