用CUDA在码头上运行火炬说模块' cutorch'未找到

时间:2017-03-09 13:32:54

标签: docker machine-learning torch

我正在运行Ubuntu 16.04。我有一个完美运行深度神经网络的docker容器。但是如果我指定使用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黑色。

1 个答案:

答案 0 :(得分:1)

想出来。我使用的是一个仅用于CPU执行的docker文件。

我应创建另一个泊坞窗文件或使用Dl-dockerNvidia-docker之类的内容。