現在,深層学習用に環境構築をしております.
複数のマシンで動かすことを想定しているのでDockerを用いようと考えており,Ubuntuの最新Dockerイメージを基にコンテナを作成しています.
具体的には
$ docker run -it --name dl ubuntu
でコンテナを作成後,
$ docker exec -it dl /bin/bash $ apt-get install pciutils $ apt-get update -y $ lspci | grep -i nvidia
でGPUを確認したのですが,何の応答もありませんでした.
Docker上のUbuntuからGPUを認識させることはできないのでしょうか?
ちなみにWindows上ではGPUを認識できており,学習もさせることはできます.
##### 環境
- Windows 10 Education
- Docker Engine Community: 19.03.08
- Ubuntu: 18.04 LTS
- GeForce RTX 2080
https://www.google.com/search?q=docker+gpu+ubuntu
この辺は参照されて構築されているという前提でいいですか?
はい.その辺を参照しつつ構築していますが,最初のGPUを確認するというところでつまずいています.
https://hub.docker.com/r/nvidia/cuda/
nvidiaからDocker imageが公開されているので、認識させることは可能です。構築手段や環境などの情報を提示してみてはいかがでしょうか。途中段階や、利用する環境自体が対応していないかもしれません。
念のためGPUのメーカー・型番を質問に追記した方がいいと思います
すみません,情報が不足していたので追記させていただきました.
やはりnvidiaのイメージをベースにした方がいいかもしれませんね.
あなたの回答
tips
プレビュー