MATLAB 2016中搭建MatConvNet執行環境(呼叫GPU執行)
阿新 • • 發佈:2019-02-18
自己的執行環境
- 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會執行一段時間,最後會輸出類似如下資訊,說明環境搭建成功。