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

Posted 2021-02-07 laowangxieboke

tags:

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

报错原因：

配置了多显卡环境，如果使用了Intel MKL作为BLAS,可能是Intel MKL的浮点数计算功能没有设置正确，可以使用安装CUDA时运行./deviceQuery时标注的0号GPU("Device 0")跑跑测试试试看

解决方法：

export CUDA_VISIBLE_DEVICES=0

然后make runtest

以上是关于安装caffe-ssd的GPU版本时候报错：BatchReindexLayerTest/2.TestGradient，where TypeParam=caffe::GPUdevice<float的主要内容，如果未能解决你的问题，请参考以下文章