配置SSD-caffe测试时出现“Check failed: error == cudaSuccess (10 vs. 0) invalid device ordinal”解决

解决方法:将ssd_pascal.py文件中gpus = "0,1,2,3"的GPU选择改为gpus = "0",后面的1,2,3都删掉即可。因为电脑只安装了一块GPU。还需要将ssd_pascal.py文件中中的batch_size = 32和accum_batch_size = 32同时改为8,不然会出现“Check failed: error == cudaSuccess (2 vs. 0)   out of memory”的错误。
 

你可能感兴趣的:(配置SSD-caffe测试时出现“Check failed: error == cudaSuccess (10 vs. 0) invalid device ordinal”解决)