![](/img/trans.png)
[英]Can I build (not run) a Windows Docker image from a Linux host?
[英]How can I use GPU in Docker image with Theano launched from Windows host?
我想在裝有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
由於我沒有路徑/dev/nvidiactl
等,因此在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)
為了運行CUDA Docker映像,您需要NVIDIA Docker 。 不幸的是,目前尚不支持Theano作為官方映像,但是您可以利用nvidia / cuda編寫自己的Dockerfile。
話雖如此,您將無法在Windows上執行此操作,因為Docker需要Linux VM,並且Windows上不支持VM GPU直通。
您可以嘗試以下圖像:
https://hub.docker.com/r/kaixhin/cuda-theano/
它需要nvidia-docker
nvidia-docker run -it kaixhin/cuda-theano
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.