docker - 如何使用 Anaconda 制作 docker 镜像并能够使用 GPU？

Question

我安装了 GPU 和 CUDA 驱动程序。我正在使用 docker hub 上提供的官方 anaconda 图像制作 docker 图像。当我在 docker 镜像中使用命令 nvidia-smi 时，它显示没有安装 CUDA DRIVER。

我应该如何制作一个 docker 图像才能将 Anaconda 与 GPU 一起使用？

标签： dockeranacondagpu

Answer 1

Docker 19.03 原生支持 NVIDIA GPU。安装标志nvidia-container-runtime并将其docker run --gpus all与 Anaconda 容器一起使用。请记住，您可以在使用 Docker 19.03 的 Windows 主机上运行具有 GPU 加速功能的 Windows 容器，但不能在 Linux 容器上运行。

资料来源：Windows 容器中的 GPU 加速