在CUDA的docker上運行火炬模塊'cutorch'找不到

Question

我使用的是Ubuntu 16.04。我有一個完美運行深層神經網絡的碼頭集裝箱。但是，如果我指定要使用CUDA，它拋出以下錯誤：

/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 

是否應CUDA/NVIDIA驅動程序被安裝在單獨的泊塢窗容器？如果是的話，那麼如何？

我正在使用GTX Geforce TITAN黑色。

Answer 1

想通了。我正在使用一個docker文件，它只能用於通過CPU執行。

我應該創建另一個碼頭文件或使用類似Dl-docker或Nvidia-docker的東西。