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