Check failed: error == cudaSuccess (74 vs. 0) misaligned address
生活随笔
收集整理的這篇文章主要介紹了
Check failed: error == cudaSuccess (74 vs. 0) misaligned address
小編覺得挺不錯的,現(xiàn)在分享給大家,幫大家做個參考.
以下摘自github hoszbh的回答?https://github.com/BVLC/caffe/issues/5729
modifying cudnn_conv_layer.cpp and aligning the address to be multiples of 32.
You can insert tow lines of code before?size_t total_max_workspace = ...?as follow:
size_t m=32;max_workspace = (max_workspace + m-1) / m * m; //align address to be multiples of m?
下面的操作可選,上面的就可以解決問題:
BTW, I think there is another bug, these lines should be put in else block:
for (int g = 0; g < (this->group_ * CUDNN_STREAMS_PER_GROUP); g++) {workspace[g] = reinterpret_cast<char *>(workspaceData)+g*max_workspace;}總結(jié)
以上是生活随笔為你收集整理的Check failed: error == cudaSuccess (74 vs. 0) misaligned address的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 如何插入8bit量化节点(tensorf
- 下一篇: 浮点卷积winograd算法