1. 程式人生 > >MATLAB 2016中搭建MatConvNet執行環境(呼叫GPU執行)

MATLAB 2016中搭建MatConvNet執行環境(呼叫GPU執行)

自己的執行環境

  • NVIDIA GeForce 940MX
  • Windows 10 64bit
  • MATLAB 2016b
  • Visual Studio 2015
  • CUDA 8.0.61 for Win10

注意

CUDA似乎只在8.0版本開始才支援VS 2015,之前用CUDA 7.5出很多問題,後來換8.0才弄好,安裝了VS 2015的同學需要注意此資訊。其他配置項可以靈活處理,留意一下MatConvNet的文件就好。

MATLAB、VS2015、CUDA幾乎都是預設安裝,就本文的範圍而言,沒有手動環境變數要求。

編譯

S1. 首先確認MATLAB確實配置了VS2015作為C和C++編譯器

mex -setup
mex -setup C++

然後根據MATLAB中的提示進行操作即可。

S2. 在MATLAB中進入下載的MatConvNet目錄中的matlab資料夾,在命令列中執行

vl_compilenn('enableGpu', true, 'cudaRoot', 'D:/CUDA/8.0/Toolkit')

最後一個引數為CUDA的路徑,根據實際情況輸入。編譯時需要一段時間,中間可能會有一些亂七八糟的警告,不必理會,只要最後出現若干“mex已經成功完成”即可。

測試

在命令列中輸入

run vl_setupnn

再輸入

vl_testnn('gpu', true)

MATLAB會執行一段時間,最後會輸出類似如下資訊,說明環境搭建成功。
這裡寫圖片描述