Check failed: error == cudaSuccess (10 vs. 0) invalid device ordinal

運行caffe例程包出現上面的錯誤Check failed: error == cudaSuccess (10 vs. 0) invalid device ordinal 這個問題的原因是設置的訓練GPU數量和真實物理機上的顯卡數量不相等! 弄清楚原因之後就很可以很愉快的解決問題了。 回到solover文件下,或者caffe的python執行文件,設置運行的GPU數量和指定GPU。 這是由於GPU
相關文章
相關標籤/搜索