Check failed: error == cudaSuccess (74 vs. 0) misaligned address的解决方法

今天训练网络的时候加入了Deconvolution Layer,结果一运行报Check failed: error == cudaSuccess (74 vs0misaligned address。

这是一个caffe的BUG,非常感谢https://github.com/BVLC/caffe/issues/5729hoszbh提供了解决方法,不然真是不知道要被困住多久了。

解决方法:

在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

猜你喜欢

转载自blog.csdn.net/Spada_k/article/details/81113944