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

/dev/nvidiactlパスなどがないので、Windowsでそれを行う方法はありますか? 他の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 を作成できます。

そうは言っても、Docker には Linux VM が必要であり、Windows では VM GPU パススルーがサポートされていないため、Windows では実行できません。

于 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 に答える