Running Fackel auf Docker mit CUDA sagt Modul 'Cutorch' nicht gefunden

Question

Ich benutze Ubuntu 16.04. Ich habe einen Docker-Container, in dem neurale Netze perfekt laufen. Aber wenn ich angeben CUDA zu verwenden, wirft es die folgenden Fehler:

/root/torch/install/bin/luajit: /root/torch/install/share/lua/5.1/trepl/init.lua:384: module 'cutorch' not found:No LuaRocks module found for cutorch 
    no field package.preload['cutorch'] 
    no file '/root/.luarocks/share/lua/5.1/cutorch.lua' 
    no file '/root/.luarocks/share/lua/5.1/cutorch/init.lua' 
    no file '/root/torch/install/share/lua/5.1/cutorch.lua' 
    no file '/root/torch/install/share/lua/5.1/cutorch/init.lua' 
    no file './cutorch.lua' 
    no file '/root/torch/install/share/luajit-2.1.0-beta1/cutorch.lua' 
    no file '/usr/local/share/lua/5.1/cutorch.lua' 
    no file '/usr/local/share/lua/5.1/cutorch/init.lua' 
    no file '/root/.luarocks/lib/lua/5.1/cutorch.so' 
    no file '/root/torch/install/lib/lua/5.1/cutorch.so' 
    no file '/root/torch/install/lib/cutorch.so' 
    no file './cutorch.so' 
    no file '/usr/local/lib/lua/5.1/cutorch.so' 
    no file '/usr/local/lib/lua/5.1/loadall.so' 
stack traceback: 
    [C]: in function 'error' 
    /root/torch/install/share/lua/5.1/trepl/init.lua:384: in function 'require' 
    /root/openface/batch-represent/main.lua:22: in main chunk 
    [C]: in function 'dofile' 
    /root/torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:145: in main chunk 
    [C]: at 0x00406670 

Should CUDA/nvidia-Treiber über den Docker Behälter separat installiert werden? Wenn ja, wie?

Ich verwende eine GTX Geforce TITAN schwarz.

Answer 1

Ich habe es herausgefunden. Ich benutzte eine Andockdatei, die nur für die Ausführung über die CPU gedacht war.

Ich sollte eine andere Docker-Datei erstellen oder etwas wie Dl-docker oder Nvidia-docker verwenden.