标签:报错 分享 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