docker - 如何在 ubuntu 上安装 tensorflow-rocm

Question

我正在尝试用我的 AMD GPU 制作 tensorflow，我已经搜索和尝试了好几天，最后我使用了 tensorflow-rocm，这会很棒（如果它有效:/）但遗憾的是我遵循了许多指南和许多说明没有结果，我尝试的最后一个教程是这个

https://github.com/RadeonOpenCompute/ROCm-docker/blob/master/quick-start.md

所有的说明都很好，除了最后一步我应该确保 docker 容器已经成功安装，我应该输入

hcc --version

但它给了我“找不到命令”

和另一个命令

/opt/rocm/bin/rocminfo

让我确保您在渲染组中，我毫无疑问！

我再次查看了 ROCm 网站，发现他们支持我的 ubuntu 20.4，但是当我有 5.11 时使用内核 5.8，当我恢复默认内核时它安装了 5.4！这可能是问题吗？

请注意，这是我第一次尝试 docker ：/如果有人知道安装 tensorflow-rocm 的更好方法，我会很高兴

标签： dockertensorflowdeep-learningamdamd-rocm