今天训练网络的时候加入了Deconvolution Layer,结果一运行报Check failed: error == cudaSuccess (74 vs. 0) misaligned address。
这是一个caffe的BUG,非常感谢https://github.com/BVLC/caffe/issues/5729的hoszbh提供了解决方法,不然真是不知道要被困住多久了。
解决方法:
在cudnn_conv_layer.cpp “ size_t total_max_workspace = ...
”前加入:
-
size_t m=32;
-
max_workspace = (max_workspace + m-1) / m * m; //align address to be multiples of m