caffe 呼叫多塊gpu

2021-08-18 16:22:59 字數 1039 閱讀 7467

一.檢測配置

使用命令檢視機器裡面安裝了幾塊顯示卡

二.更改配置檔案

在檢視到機器裡面安裝的顯示卡不止一塊時,就可以通過更改配置檔案,在編譯通過後,就可以使用多個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...