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

Posted laowangxieboke

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了安装caffe-ssd的GPU版本时候报错:BatchReindexLayerTest/2.TestGradient,where TypeParam=caffe::GPUdevice<doubl相关的知识,希望对你有一定的参考价值。

报错原因:使用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<doubl的主要内容,如果未能解决你的问题,请参考以下文章

caffe-ssd的GPU在make runtest的时候报错:BatchReindexLayerTest/2.TestGradient,where TypeParam=caffe::GPUdevic

detectron2安装出现Kernel not compiled with GPU support 报错信息

informatic在安装服务端的时候,报错安装程序无法创建域,缺失infasetup.bat。

win10 python3.5 tensorflow(gpu)安装

解决Google colab上安装GPU版本mxnet报错:libnvrtc.so.11.2: cannot open shared object file: No such file...

如何安装dlib gpu版本