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 的安装位置;