如何通过从 Windows 主机启动的 Theano 在 Docker 映像中使用 GPU？答案

【问题标题】：How can I use GPU in Docker image with Theano launched from Windows host?如何通过从 Windows 主机启动的 Theano 在 Docker 映像中使用 GPU？
【发布时间】：2016-03-22 10:36:34
【问题描述】：

我想在安装了 Windows 的 PC 上通过 Docker 映像运行 Theano。 Docker 映像包含 Ubuntu 系统、CUDA 驱动程序和 Theano (https://hub.docker.com/r/kaixhin/cuda-theano/)，但为了在我的算法中使用 GPU，我需要将 Nvidia 设备附加到映像：

docker run -it --device /dev/nvidiactl --device /dev/nvidia-uvm --device /dev/nvidia0 kaixhin/cuda-theano

有没有办法在 Windows 中做到这一点，因为我没有路径 /dev/nvidiactl 等？我一直在寻找其他Docker images，但似乎所有这些都使用Linux 作为主机系统。是否有允许我从 Windows 使用 GPU 的版本？

现在我可以在 Docker 中运行我的脚本，但它只使用我的 CPU：

WARNING (theano.sandbox.cuda): CUDA is installed, but device gpu is not available (error: Unable to get the number of gpus available: no CUDA-capable device is detected)

【问题讨论】：

标签： windows docker theano-cuda

【解决方案1】：

要运行 CUDA Docker 映像，您需要 NVIDIA Docker。不幸的是，Theano 目前不支持作为官方镜像，但您可以使用 nvidia/cuda 编写自己的 Dockerfile

话虽如此，您将无法在 Windows 上执行此操作，因为 Docker 需要 Linux VM，并且在 Windows 上不支持 VM GPU 直通。

【讨论】：

【解决方案2】：

你可以试试这张图片：

https://hub.docker.com/r/kaixhin/cuda-theano/

需要nvidia-docker

nvidia-docker run -it kaixhin/cuda-theano

【讨论】：