radeon 560 上 Docker 主机 运行 上的 tensorflow-gpu?
tensorflow-gpu on a Docker of host running on radeon 560?
我正在尝试将 tensorflow 用于深度学习,但对于 GPU 支持,tensorflow 需要 Nvidia GPU,但我有 Radeon 560。在 Docker 中有 运行 tensorflow GPU 的任何方式吗容器?我无法理解 Docker。除了使用 Colab 或 Docker 之外,还有其他可能性吗?
目前官方的tensorflow-gpu docker images 只支持NVIDIA gpus 和host 运行 NVIDIA cuda-driver。要使用 AMD GPU,您需要使用支持 ROCm 的 TensorFlow 版本。 Google for "tensorflow ROCm" 你会找到一些关于如何做到这一点的页面。最好的link大概是:
我正在尝试将 tensorflow 用于深度学习,但对于 GPU 支持,tensorflow 需要 Nvidia GPU,但我有 Radeon 560。在 Docker 中有 运行 tensorflow GPU 的任何方式吗容器?我无法理解 Docker。除了使用 Colab 或 Docker 之外,还有其他可能性吗?
目前官方的tensorflow-gpu docker images 只支持NVIDIA gpus 和host 运行 NVIDIA cuda-driver。要使用 AMD GPU,您需要使用支持 ROCm 的 TensorFlow 版本。 Google for "tensorflow ROCm" 你会找到一些关于如何做到这一点的页面。最好的link大概是: