CUDA是NVIDIA推出的用于自家GPU的并行计算框架,也就是说CUDA只能在NVIDIA的GPU上运行,而且只有当要解决的计算问题是可以大量并行计算的时候才能发挥CUDA的作用。
cuDNN是一个常见的神经网络层加速库文件,其能够很大程度的把加载到显卡上的网络层数据进行优化计算,cuDNN需要在有cuda的基础上进行。
一、查看电脑显卡对应CUDA版本
二、下载CUDA
https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
三、安装CUDA
安装CUDA时选择自定义安装,并将Visual Studio Integration一项去掉打勾,否则有可能安装出错
四、检查是否安装正确
检验安装是否安装正确:win+R→cmd→nvcc -V
五、下载cuDNN
https://developer.nvidia.com/rdp/cudnn-archive
将cuDNN解压后的文件夹中bin、include和lib三个文件夹
把他们复制粘贴到CUDA安装路径中这三个文件夹的父文件夹下面,Windows中默认安装路径为:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0