Multi-GPU Usage:caffe下的多GPU同時使用

最近在跑FCN的時候發現caffe.set_device()只能有一個int參數,導致只能用一個GPU跑,於是開始着手解決。

caffe聲明:https://github.com/BVLC/caffe/blob/master/docs/multigpu.md

截圖如下:


多GPU接口只支持C/C++,以及訓練。

正在絞盡腦汁解決~

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章