search

soumith / imagenet-multiGPU.torch

an imagenet example in torch.
BSD 2-Clause "Simplified" License
401 stars 158 forks source link

utils.lua nn.DataParallelTable requires 'cunn' #60

Closed anuragranj closed 8 years ago

anuragranj commented 8 years ago

With nGPU>1, the following error occurs on running th main.lua

[string "model = nn.DataParallelTable(1)"]:1: attempt to call field 'DataParallelTable' (a nil value)
stack traceback:
[string "model = nn.DataParallelTable(1)"]:1: in main chunk
[C]: in function 'xpcall'
/home/aranjan/torch/install/share/lua/5.1/trepl/init.lua:669: in function 'repl'
...njan/torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:199: in main chunk
[C]: at 0x00406670

This is because nn.DataParallelTable requires cunn. The following following test summarizes the issue and a possible fix.

th> model = nn.DataParallelTable(1)
[string "model = nn.DataParallelTable(1)"]:1: attempt to call field 'DataParallelTable' (a nil value)
stack traceback:
[string "model = nn.DataParallelTable(1)"]:1: in main chunk
[C]: in function 'xpcall'
/home/aranjan/torch/install/share/lua/5.1/trepl/init.lua:669: in function 'repl'
...njan/torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:199: in main chunk
[C]: at 0x00406670  

th> require 'cunn'
true    
                                                                  [0.0397s] 
th> model = nn.DataParallelTable(1)
                                                                  [0.0001s]