cuda runtime error when training

[chengjunjiecn]

Hi,

when I run,
th train.lua -h5_file /data/dataset.h5 -loss_network /data/models/vgg16.t7 -style_image /data/images/styles/wave.jpg -style_image_size 384 -content_weights 1.0 -style_weights 5.0 -checkpoint_name checkpoint -gpu 0

I meet the issue.
my cuda is 9.0 and cudaa is V7.
Can you help? thanks.

THCudaCheck FAIL file=/tmp/luarocks_cutorch-scm-1-7592/cutorch/lib/THC/generic/THCTensorMath.cu line=15 error=48 : no kernel image is available for execution on the device

/root/torch/install/bin/luajit: /root/torch/install/share/lua/5.1/nn/Container.lua:67:

In 1 module of nn.Sequential:

/root/torch/install/share/lua/5.1/nn/THNN.lua:110: cuda runtime error (48) : no kernel image is available for execution on the device at /tmp/luarocks_cutorch-scm-1-7592/cutorch/lib/THC/generic/THCTensorMath.cu:15

stack traceback:

[C]: in function 'v'

/root/torch/install/share/lua/5.1/nn/THNN.lua:110: in function 'SpatialConvolutionMM_updateOutput'

/root/torch/install/share/lua/5.1/nn/SpatialConvolution.lua:79: in function </root/torch/install/share/lua/5.1/nn/SpatialConvolution.lua:76>

[C]: in function 'xpcall'

/root/torch/install/share/lua/5.1/nn/Container.lua:63: in function 'rethrowErrors'

/root/torch/install/share/lua/5.1/nn/Sequential.lua:44: in function 'forward'

./fast_neural_style/PerceptualCriterion.lua:78: in function 'setStyleTarget'

train.lua:134: in function 'main'

train.lua:327: in main chunk

[C]: in function 'dofile'

/root/torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:150: in main chunk

[C]: at 0x00405d50

WARNING: If you see a stack trace below, it doesn't point to the place where this error occurred. Please use only the one above.

stack traceback:

[C]: in function 'error'

/root/torch/install/share/lua/5.1/nn/Container.lua:67: in function 'rethrowErrors'

/root/torch/install/share/lua/5.1/nn/Sequential.lua:44: in function 'forward'

./fast_neural_style/PerceptualCriterion.lua:78: in function 'setStyleTarget'

train.lua:134: in function 'main'

train.lua:327: in main chunk

[C]: in function 'dofile'

/root/torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:150: in main chunk

[C]: at 0x00405d50