tensorflow gpu在radeon 560上运行的主机Docker上

Question

我正在尝试tensorflow用于深度学习，但对于GPU支持，tensorflow需要Nvidia GPU，但我有Radeon 560。有没有办法在Docker容器中运行tensorflow GPU？我无法理解Docker。除了使用Colab或Docker，还有其他可能性吗？

Answer 1

此时的官方tensorflow gpu docker镜像仅支持NVIDIA gpu和运行NVIDIA cuda驱动程序的主机。为了使用AMD GPU，您需要使用支持ROCm的TensorFlow版本。在谷歌上搜索"tensorflow ROCm"，你会发现一些关于如何做到这一点的页面。最好的链接可能是：

https://github.com/ROCmSoftwarePlatform/tensorflow-upstream