本文以cuda8.0和cudnn5.0安装为例,显卡为GTX1060。
本文主要介绍cuda与cudnn环境配置过程中,存在着的一些需要注意的版本方面的问题和链接问题,详细安装流程等请自行百度。
先简单介绍一下cuda以及cudnn。
cuda是NVIDIA推出的GPU的运算引擎,配合NVIDIA显卡使用,对于不同的NVIDIA,cuda兼容版本也不同,高版本的卡不支持低版本的cuda,同时低端卡安装高版本的cuda也会出现问题。查询显卡支持的cuda版本可以前往NVIDIA官网链接https://developer.nvidia.com/cuda-gpus,或者在windows下查看NVIDIA控制面板中的系统信息。详情见百度链接:https://jingyan.baidu.com/article/6fb756ec4fabc4241858fbf7.html
cudnn是神经网络GPU的加速库,对于一些网络框架,如caffe,使用cudnn能够进一步提升caffe的运算速度,但并非必须的支持,缺少cudnn也可以运行caffe,只不过速度稍微慢点而已。但对于依赖cudnn的神经网络框架,如tensorflow,则是必须的。
cuda安装流程不详细介绍,注意的一点是安装过程中选择是否安装NVIDIA驱动时,拒绝安装。其余accept或者default就OK。安装完成后在环境变量中加入以下两项(不同版本cuda请注意更换版本号)。