torch 平台装下,gup测试neural-style

ybl0057 2018-01-29 09:28:21
ybl@ybl-Lenovo-ideapad-Y700-15ISK:~/neural-style$ th neural_style.lua -gpu 0 -print_iter 1
[libprotobuf WARNING google/protobuf/io/coded_stream.cc:537] Reading dangerously large protocol message. If the message turns out to be larger than 1073741824 bytes, parsing will be halted for security reasons. To increase the limit (or to disable these warnings), see CodedInputStream::SetTotalBytesLimit() in google/protobuf/io/coded_stream.h.
[libprotobuf WARNING google/protobuf/io/coded_stream.cc:78] The total number of bytes read was 574671192
Successfully loaded models/VGG_ILSVRC_19_layers.caffemodel
conv1_1: 64 3 3 3
conv1_2: 64 64 3 3
conv2_1: 128 64 3 3
conv2_2: 128 128 3 3
conv3_1: 256 128 3 3
conv3_2: 256 256 3 3
conv3_3: 256 256 3 3
conv3_4: 256 256 3 3
conv4_1: 512 256 3 3
conv4_2: 512 512 3 3
conv4_3: 512 512 3 3
conv4_4: 512 512 3 3
conv5_1: 512 512 3 3
conv5_2: 512 512 3 3
conv5_3: 512 512 3 3
conv5_4: 512 512 3 3
fc6: 1 1 25088 4096
fc7: 1 1 4096 4096
fc8: 1 1 4096 1000
Setting up style layer 2 : relu1_1
Setting up style layer 7 : relu2_1
Setting up style layer 12 : relu3_1
Setting up style layer 21 : relu4_1
Setting up content layer 23 : relu4_2
Setting up style layer 30 : relu5_1
Capturing content targets
nn.Sequential {
[input -> (1) -> (2) -> (3) -> (4) -> (5) -> (6) -> (7) -> (8) -> (9) -> (10) -> (11) -> (12) -> (13) -> (14) -> (15) -> (16) -> (17) -> (18) -> (19) -> (20) -> (21) -> (22) -> (23) -> (24) -> (25) -> (26) -> (27) -> (28) -> (29) -> (30) -> (31) -> (32) -> (33) -> (34) -> (35) -> (36) -> (37) -> output]
(1): nn.TVLoss
(2): nn.SpatialConvolution(3 -> 64, 3x3, 1,1, 1,1)
(3): nn.ReLU
(4): nn.StyleLoss
(5): nn.SpatialConvolution(64 -> 64, 3x3, 1,1, 1,1)
(6): nn.ReLU
(7): nn.SpatialMaxPooling(2x2, 2,2)
(8): nn.SpatialConvolution(64 -> 128, 3x3, 1,1, 1,1)
(9): nn.ReLU
(10): nn.StyleLoss
(11): nn.SpatialConvolution(128 -> 128, 3x3, 1,1, 1,1)
(12): nn.ReLU
(13): nn.SpatialMaxPooling(2x2, 2,2)
(14): nn.SpatialConvolution(128 -> 256, 3x3, 1,1, 1,1)
(15): nn.ReLU
(16): nn.StyleLoss
(17): nn.SpatialConvolution(256 -> 256, 3x3, 1,1, 1,1)
(18): nn.ReLU
(19): nn.SpatialConvolution(256 -> 256, 3x3, 1,1, 1,1)
(20): nn.ReLU
(21): nn.SpatialConvolution(256 -> 256, 3x3, 1,1, 1,1)
(22): nn.ReLU
(23): nn.SpatialMaxPooling(2x2, 2,2)
(24): nn.SpatialConvolution(256 -> 512, 3x3, 1,1, 1,1)
(25): nn.ReLU
(26): nn.StyleLoss
(27): nn.SpatialConvolution(512 -> 512, 3x3, 1,1, 1,1)
(28): nn.ReLU
(29): nn.ContentLoss
(30): nn.SpatialConvolution(512 -> 512, 3x3, 1,1, 1,1)
(31): nn.ReLU
(32): nn.SpatialConvolution(512 -> 512, 3x3, 1,1, 1,1)
(33): nn.ReLU
(34): nn.SpatialMaxPooling(2x2, 2,2)
(35): nn.SpatialConvolution(512 -> 512, 3x3, 1,1, 1,1)
(36): nn.ReLU
(37): nn.StyleLoss
}
/home/ybl/torch/install/bin/luajit: /home/ybl/torch/install/share/lua/5.1/nn/Container.lua:67:
In 2 module of nn.Sequential:
/home/ybl/torch/install/share/lua/5.1/nn/THNN.lua:110: cublas runtime error : library not initialized at /tmp/luarocks_cutorch-scm-1-7236/cutorch/lib/THC/THCGeneral.c:405
stack traceback:
[C]: in function 'v'
/home/ybl/torch/install/share/lua/5.1/nn/THNN.lua:110: in function 'SpatialConvolutionMM_updateOutput'
...bl/torch/install/share/lua/5.1/nn/SpatialConvolution.lua:79: in function <...bl/torch/install/share/lua/5.1/nn/SpatialConvolution.lua:76>
[C]: in function 'xpcall'
/home/ybl/torch/install/share/lua/5.1/nn/Container.lua:63: in function 'rethrowErrors'
/home/ybl/torch/install/share/lua/5.1/nn/Sequential.lua:44: in function 'forward'
neural_style.lua:162: in function 'main'
neural_style.lua:601: in main chunk
[C]: in function 'dofile'
.../ybl/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'
/home/ybl/torch/install/share/lua/5.1/nn/Container.lua:67: in function 'rethrowErrors'
/home/ybl/torch/install/share/lua/5.1/nn/Sequential.lua:44: in function 'forward'
neural_style.lua:162: in function 'main'
neural_style.lua:601: in main chunk
[C]: in function 'dofile'
.../ybl/torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:150: in main chunk
[C]: at 0x00405d50

这是测试neural-style的代码,然后就报错了 library not initialized,不能跑GPU,可以跑CUP的,之前已经装过caffe和CUDA,这是为什么呀?
...全文
1085 1 打赏 收藏 转发到动态 举报
写回复
用AI写文章
1 条回复
切换为时间正序
请发表友善的回复…
发表回复

4,436

社区成员

发帖
与我相关
我的任务
社区描述
Linux/Unix社区 内核源代码研究区
社区管理员
  • 内核源代码研究区社区
加入社区
  • 近7日
  • 近30日
  • 至今
社区公告
暂无公告

试试用AI创作助手写篇文章吧