docker - 如何在 Docker 中使用 GPU 重新训练对象检测模型？

Question

我一直在关注this谷歌珊瑚关于在 docker 中重新训练对象检测模型的教程，它明确指出这仅用于 CPU 训练，这非常慢。
有没有一种简单的方法可以移植这个 docker 容器以利用 GPU(nvidia GTX 1080)。我已经安装了 nvidia-docker2，并成功地将我的 gpu 传递到其他容器中，据我所知，也是这个容器，使用 --gpus all标签。 nvidia-smi命令在我的容器中工作，所以我几乎可以肯定我的 GPU 已经成功通过，但是在训练模型时没有使用它。
根据nvidia-smi，容器内部和外部的CUDA版本是11.4，我使用的是Ubuntu 20.04。

Answer 1

您可以尝试使用 Docker https://github.com/google-coral/tutorials/issues/5#issuecomment-821860067 提供的解决方案。或使用 google-coral 教程 (https://github.com/google-coral/tutorials) 中基于 GPU 的 colab 教程重新训练对象检测模型