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

但是SSD配置時(shí)的例子是將訓(xùn)練語句整合成一個(gè)python文件ssd_pascal.py，所以需要改此代碼。相關(guān)配置訓(xùn)練方法請(qǐng)參看轉(zhuǎn)載博文：http://blog.csdn.net/xunan003/article/details/78427446

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

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

總結(jié)：這兩個(gè)錯(cuò)誤只是由于博主硬件而出現(xiàn)的，所以請(qǐng)各位讀者根據(jù)自己的硬件條件加以修正。

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。