caffe運行錯誤: im2col.cu:61] Check failed: error == cudaSuccess (8 vs. 0) invalid device function


錯誤

 im2col.cu:61] Check failed: error == cudaSuccess (8 vs. 0)  invalid device function

原因:由於Makefile.config里面只對cuda8.0一下的架構編譯cuda程序,所以采用cuda8.0后cuda程序沒有被編譯

解決:修改Makefile.config中的 CUDA_ARCH

添加 -gencode arch=compute_61,code=sm_61

計算能力可以參考下面鏈接添加合適的架構參數

http://blog.csdn.net/allyli0022/article/details/54628987

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM