0

我想在安装了 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 映像，但似乎所有这些都使用 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)

4

2 回答 2

1

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

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

于 2016-03-23T00:20:27.880 回答
0

你可以试试这张图片：

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

这个需要nvidia-docker

nvidia-docker run -it kaixhin/cuda-theano

于 2017-01-17T03:48:54.753 回答