运行与CUDA的docker上的火炬说没有find模块“cutorch”

我正在运行Ubuntu 16.04。 我有一个docker集装箱运行深度neural network完美。 但是,如果我指定使用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驱动程序分别安装在Docker容器上吗? 如果是,那么怎么样?

我正在使用GTX Geforce TITAN黑色。

弄清楚了。 我正在使用一个docker文件,它只是用来执行CPU。

我应该创build另一个docker文件或使用像Dl-docker或者Nvidia-docker这样的东西 。