利用caffe的MATLAB接口跑深度學習時，設置gpu模式：caffe.set_mode_gpu()，可以加速運算，然而在MATLAB 2016a/b中調用gpu時會出現了一個BUG，這個BUG會導致運算速度巨慢無比，正常gpu模式需要10s的可能要跑到10min，簡直比cpu模式還慢，不能忍！

查找了半天最終在MathWorks 的Bug Reports(1439741)中發現解決辦法，以下是bug描述及鏈接(需要注冊)：

https://www.mathworks.com/support/bugreports/search_results?utf8=%E2%9C%93&search_executed=1&keyword=&release_filter=Exists+in&release=264&selected_products=162&commit=

Search%20%20%E4%B8%8B%E9%9D%A2%E6%98%AFbug%E6%8F%8F%E8%BF%B0%EF%BC%9A

簡而言之，就是對于最新的compute capability 6.0以上的顯卡(gtx1080 gtx1070 gtx1060等)，MATLAB 目前為止采用的cuDNN庫(matlabroot\bin\win64\cudnn64_4.dll)支持的不好，有bug。需要下載下面這個文件，解壓之后將matlabroot\bin\win64\cudnn64_4.dll替換掉：

從上到下分別對應著linux、windows的16a、16b版本，以下是MATLAB 2016aforwindows 的下載鏈接(需要登錄MATLAB賬戶)：

https://www.mathworks.com/support/bugreports/download/7370

替換之后，還需要設置一下環境變量：

重啟一下電腦，跑一下MTCNN，速度是飛一般的快，至此，問題圓滿解決。

總結

以上是生活随笔為你收集整理的matlab2016b慢,Matlab 2016a/b中调用GPU速度巨慢的解决办法的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。