如何在Windows主机上启动Theano的Docker镜像中使用GPU?

时间:2016-03-22 10:36:34

标签: windows docker theano-cuda

我希望在安装了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)

2 个答案:

答案 0 :(得分:1)

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

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

答案 1 :(得分:0)

您可以尝试此图片:

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

需要nvidia-docker

nvidia-docker run -it kaixhin/cuda-theano