MathWorks - Bug Reports
1. 問題說明
當運作 alexnet 等卷積神經網絡需要使用 GPU 加速時,matlab 如果提示如下的警告資訊:
GPUs of compute capability 6.0 and above are not supported due
to a problem with this version of the cuDNN deep learning library.
See bug report 1439741. Switching to CPU.
這說明,GPU 的版本(計算能力 compute capability)與卷積神經網絡實作的相容性出現了問題,此時通過 GPU 加速程式的運作不再可行,而轉向在較為低速的 CPU 運作。
>> D = gpuDevice;
>> D.ComputeCapability
ans =
6.1
gpuDevice
函數擷取對目前主機的顯示卡資訊的描述,通過其下的 ComputeCapability 屬性可檢視目前計算性能的版本,>= 6.0 版本的 GPU 将出現和卷積神經網絡不相容的狀況發生。
2. 解決方案
點選進入 MathWorks - Bug Reports :
- 下載下傳與自己系統以及matlab 版本相比對的附件(attachment);
- 解壓檔案;
- 将解壓後的檔案覆寫到 matlab 的安裝位置;