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

转载于:https://www.cnblogs.com/mtcnn/p/9421908.html

你可能感兴趣的:(【matlab】GPU 显卡版本与计算能力(compute capability)兼容性问题)