码迷,mamicode.com
首页 > 其他好文 > 详细

安装caffe-ssd的GPU版本时候报错:BatchReindexLayerTest/2.TestGradient,where TypeParam=caffe::GPUdevice<double>

时间:2019-02-01 20:19:10      阅读:522      评论:0      收藏:0      [点我收藏+]

标签:报错   分享   ice   9.1   png   文件   方法   com   bsp   

报错原因:使用CUDA 9.1加双显卡

解决方法:直接将BatchReindexLayerTest的forWord和backWord改为CPU执行

具体来说就是将~/caffe/src/caffe/layers/batch_reindex_layer.cu文件内的

技术分享图片

 

安装caffe-ssd的GPU版本时候报错:BatchReindexLayerTest/2.TestGradient,where TypeParam=caffe::GPUdevice<double>

标签:报错   分享   ice   9.1   png   文件   方法   com   bsp   

原文地址:https://www.cnblogs.com/laowangxieboke/p/10346869.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!