一.檢測配置
使用命令檢視機器裡面安裝了幾塊顯示卡
二.更改配置檔案
在檢視到機器裡面安裝的顯示卡不止一塊時,就可以通過更改配置檔案,在編譯通過後,就可以使用多個gpu同時執行。
設定makefile.config
開啟開關: use_nccl := 1, 並新增nccl庫路徑
use_nccl
:= 1
#將之前的注釋去掉,開啟開關。
include_dirs += /path/nccl
/build/include
#在原有的路徑下方加上這句話即可
library_dirs += /path/nccl
/build/lib
三.安裝nccl
git clone
cd nccl
sudo make install
sudo gedit ~/.bashrc
export ld_library_path="$ld_library_path:/home/***/nccl/build/lib/"
#***為使用者名稱
source .bashrc
在安裝好以後,我們重新編譯caffe即可
cd caffe
sudo make all
sudo make test
sudo make pycaffe
sudo make runtest
四.使用多個gpu
我們在使用caffe的工具訓練網路時,在語句的最後加上以下內容即可
-gpu all #執行所有的gpu
-gpu 1,2
#執行1號2號gpu
caffe和tensorflow對GPU的使用
方法一 在solver檔案中寫入 solver mode cpu or gpu solver mode gpu方法二 命令列引數 gpu train build tools caffe train solver examples mnist lenet solver.prototxt gpu 0 t...
Mac下Caffe安裝 無GPU
系統版本 os x 10.11.2 處理器 2.2ghz i7 記憶體 16gb 由於caffe安裝過程需要依賴多個包,因此最好提前安裝好包管理工具。在mac上推薦使用homebrew,如果是mac新手可參考高效 macbook 工作環境配置 的開發環境配置部分。ruby e curl fssl 注...
Opencv呼叫GPU模組
1 opencv提供的開發包中提供的庫沒有開啟gpu和ocl模組功能,雖然有 gpu.lib gpu.dll檔案,但不能用。如果呼叫gpu getcudaenabledevicecount 將會return 0 要開啟該功能需要重新編譯opencv的庫。2 參考和文章,需要在編譯之前安裝 cmake...