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

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

时间:2019-02-01 20:16:08      阅读:438      评论:0      收藏:0      [点我收藏+]

标签:exp   解决   ram   device   HERE   float   query   caff   test   

报错原因:

配置了多显卡环境,如果使用了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>

标签:exp   解决   ram   device   HERE   float   query   caff   test   

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

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