天天看點

【matlab】GPU 顯示卡版本與計算能力(compute capability)相容性問題

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 的安裝位置;
【matlab】GPU 顯示卡版本與計算能力(compute capability)相容性問題