报错原因:使用CUDA 9.1加双显卡
解决方法:直接将BatchReindexLayerTest的forWord和backWord改为CPU执行
具体来说就是将~/caffe/src/caffe/layers/batch_reindex_layer.cu文件内的
原文地址:https://www.cnblogs.com/laowangxieboke/p/10346869.html
时间: 2024-11-09 06:16:00
报错原因:使用CUDA 9.1加双显卡
解决方法:直接将BatchReindexLayerTest的forWord和backWord改为CPU执行
具体来说就是将~/caffe/src/caffe/layers/batch_reindex_layer.cu文件内的
原文地址:https://www.cnblogs.com/laowangxieboke/p/10346869.html