簡體   English   中英

從Windows主機啟動的Theano中,如何在Docker映像中使用GPU?

[英]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.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM