如何确定PC Nvidia显卡是否支持CUDA以及cudaNN?

Nvidia显卡几乎已经垄断了当前火热的神经网络训练领域,cudaNN是NVIDIA CUDA 深度神经网络库 (cuDNN) ,而CUDA则是基于Nvidia的统一计算平台,界各地的深度学习研究人员和框架开发者都依赖 cuDNN 实现高性能 GPU 加速。借助 cuDNN,研究人员和开发者可以专注于训练神经网络及开发软件应用,而不必花时间进行低层级的 GPU 性能调整。cuDNN 可加速广泛应用的深度学习框架,包括 Caffe2、Chainer、Keras、MATLAB、MxNet、PyTorch 和 TensorFlow。既然这么有用,那么如何确定你的PC是否支持N卡的网络加速平台引擎CUDA以及cudaNN呢?下面介绍步骤:

首先确认你是否有N卡,就拿我的破电脑为例,打开设备管理器,可以看到PC上安装了Nvidia 的MX250独显,没听说过没关系,总之你知道是很弱很弱的就得了。

如何确定PC Nvidia显卡是否支持CUDA以及cudaNN?_第1张图片

打开N卡控制面板:

如何确定PC Nvidia显卡是否支持CUDA以及cudaNN?_第2张图片

 帮助-》系统信息,可以看到系统上支持的是DirectX12.0版本。

如何确定PC Nvidia显卡是否支持CUDA以及cudaNN?_第3张图片

 组件,3D设置栏目中,你将会看当前显卡支持的CUDA版本。

如何确定PC Nvidia显卡是否支持CUDA以及cudaNN?_第4张图片

下载cuda:

访问网站https://developer.nvidia.com/cuda-toolkit-archive

如何确定PC Nvidia显卡是否支持CUDA以及cudaNN?_第5张图片

 选择正确的版本,点击进入:

如何确定PC Nvidia显卡是否支持CUDA以及cudaNN?_第6张图片

 然后选择对应的平台下载:

如何确定PC Nvidia显卡是否支持CUDA以及cudaNN?_第7张图片

cuNN下载:

地址:https://developer.nvidia.com/zh-cn/cudnn

如何确定PC Nvidia显卡是否支持CUDA以及cudaNN?_第8张图片

 下载CUNN需要注册登录,填写一系列的表单,一顿操作完之后,即可见到CUNN下载窗口:

如何确定PC Nvidia显卡是否支持CUDA以及cudaNN?_第9张图片

点击正确的版本和平台下载即可。

如何确定PC Nvidia显卡是否支持CUDA以及cudaNN?_第10张图片 安装比较简单:

如何确定PC Nvidia显卡是否支持CUDA以及cudaNN?_第11张图片

如何确定PC Nvidia显卡是否支持CUDA以及cudaNN?_第12张图片

 安装CUDA IDE

如何确定PC Nvidia显卡是否支持CUDA以及cudaNN?_第13张图片

如何确定PC Nvidia显卡是否支持CUDA以及cudaNN?_第14张图片

而CUNN则是一系列的库的集合:

如何确定PC Nvidia显卡是否支持CUDA以及cudaNN?_第15张图片

头文件:

如何确定PC Nvidia显卡是否支持CUDA以及cudaNN?_第16张图片

如何确定PC Nvidia显卡是否支持CUDA以及cudaNN?_第17张图片

结束!

你可能感兴趣的:(人工智能,Linux,嵌入式系统,深度学习,神经网络,pytorch)