I want to run Theano via Docker image on my PC with Windows installed. The Docker image contains Ubuntu system, CUDA drivers and Theano ( https://hub.docker.com/r/kaixhin/cuda-theano/ ) but in order to use GPU in my algorithm I need to attach Nvidia devices to the image:
docker run -it --device /dev/nvidiactl --device /dev/nvidia-uvm --device /dev/nvidia0 kaixhin/cuda-theano
Is there a way to do it in Windows, since I don't have a path /dev/nvidiactl
etc.? I have been looking for other Docker images but it seems that all of these are using Linux as the host system. Is there a version that will allow me to use GPU from Windows?
For now I can run my script in Docker, but it uses only my 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)
In order to run CUDA Docker images you need NVIDIA Docker . Unfortunately, Theano is not supported as an official image at the moment but you can write your own Dockerfile leveraging nvidia/cuda
Having said that, you won't be able to do it on Windows because Docker needs a Linux VM and there is no support for VM GPU passthrough on Windows.
You can try this image:
https://hub.docker.com/r/kaixhin/cuda-theano/
It requires nvidia-docker
nvidia-docker run -it kaixhin/cuda-theano
The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.