這是由於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 ...”的錯誤。
---------------------
作者：xunan003
來源：CSDN
原文：https://blog.csdn.net/xunan003/article/details/78433383
版權宣告：本文為博主原創文章，轉載請附上博文連結！

ubuntu14.04 caffe 執行別人的程式，報錯Check failed: error == cudaSuccess (10 vs. 0)  invalid device ordinal。根據網上的說法，GPU記憶體不夠，可是batch大小改為1了還是不行。問了博士師兄才發現，在該行上面有一行，寫著using GPUs 1.而我只有一個GPU，如圖2，我把solver_48.prototxt中最後一行的device_id改為0，接著執行，見圖，這個錯誤就沒有啦～～