今天訓練網絡的時候加入了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

總結

以上是生活随笔為你收集整理的Check failed: error == cudaSuccess (74 vs. 0) misaligned address的解决方法的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。