由於最近在學卷積神經網絡,而且對matlab熟悉一些,所以就選用matlab來作爲學習卷積神經網絡的工具。首先本文假設你的機器已經安裝好了cuda,cudnn,vs2015。本文參考了博客https://blog.csdn.net/u011935830/article/details/82187321。
1.MatConvNet的下載地址:https://github.com/vlfeat/matconvnet
2.下載之後將壓縮包解壓到一個文件夾,並在這個文件夾的matlab文件夾下新建local文件夾,把cudnn放入這個文件夾,然後matlab的當前文件夾目錄進入這個文件夾的matlab文件夾
3.在命令行窗口輸入:
mex -setup
4.如上圖點擊選擇畫紅框部分
5.在命令行窗口輸入:
vl_compilenn('enableGpu', true,... 'cudaRoot', 'C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0', ... %自己安裝的CUDA的路徑 'cudaMethod', 'nvcc', ... 'enableCudnn', true, ... 'cudnnRoot', ... 'E:\matlab2018\toolbox\matconvnn\matconvnetmaster\matlab\local\cuda') ;
6..對vl_compilenn.m中的代碼按照https://blog.csdn.net/u014292102/article/details/80331481(這一步是對出現的以下圖片錯誤時才執行,否則請跳過)
7.最後全部編譯成功: