這是由于GPU數量不匹配造成的，如果訓練自己的數據，那么我們只需要將solver.prototxt文件中的device_id項改為自己的GPU塊數，一塊就是0，兩塊就是1，以此類推。

但是SSD配置時的例子是將訓練語句整合成一個python文件ssd_pascal.py，所以需要改此代碼。相關配置訓練方法請參看轉載博文：http://blog.csdn.net/xunan003/article/details/78427446

解決方法：將ssd_pascal.py文件中第332行gpus = "0，1，2，3"的GPU選擇改為gpus = "0"，后面的1，2，3都刪掉即可。再次訓練即可。

當然，由于博主只有一塊GPU且電腦運行內存有限，還需要將ssd_pascal.py文件中的337行batch_size = 32和338行accum_batch_size = 32都改小一倍，即更改批量大小，不然會出現“Check failed: error == cudaSuccess (2 vs. 0)? invalid ...”的錯誤。

總結：這兩個錯誤只是由于博主硬件而出現的，所以請各位讀者根據自己的硬件條件加以修正。

總結

以上是生活随笔為你收集整理的配置SSD-caffe测试时出现“Check failed: error == cudaSuccess (10 vs. 0) invalid device ordinal”解决的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。