编译caffe进行make runtest遇到caffe error == cudaSuccess (48 vs. 0)的问题

这个问题发现网上的解决办法不多,试了各种方法都不能解决,折磨本人两天多的时间,问题基本确认在Makefile.config的CUDA_ARCH配置上,最后有效找到一个解决方法,但是本人对CUDA不是很熟,不清楚其解决原因。

针对的板卡平台为Jetson TX1和Jetson Nano,cuda分别为CUDA9.0和CUDA10.0。

报错原因如下:

解决办法:

在Makefile.config中进行修改,添加如下一行:

保存后记得先make clean,再进行make all -j4。

为什么添加这么一行,是因为我发现TX1和Nano的GPU的算力能力都是5.3,所以这么做了,结果发现可以解决这个错误,哈哈。

这个报错应该和GPU的算力能力有关系,注意先到官网查看自己GPU平台算力再做相应修改。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章