win10安装cuda和cudnn

CUDA是NVIDIA推出的用于自家GPU的并行计算框架,也就是说CUDA只能在NVIDIA的GPU上运行,而且只有当要解决的计算问题是可以大量并行计算的时候才能发挥CUDA的作用。

cuDNN是一个常见的神经网络层加速库文件,其能够很大程度的把加载到显卡上的网络层数据进行优化计算,cuDNN需要在有cuda的基础上进行。

一、查看电脑显卡对应CUDA版本

win10安装cuda和cudnn
win10安装cuda和cudnn

二、下载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一项去掉打勾,否则有可能安装出错
win10安装cuda和cudnn

四、检查是否安装正确

检验安装是否安装正确:win+R→cmd→nvcc -V
win10安装cuda和cudnn

五、下载cuDNN

https://developer.nvidia.com/rdp/cudnn-archive

将cuDNN解压后的文件夹中bin、include和lib三个文件夹
win10安装cuda和cudnn
把他们复制粘贴到CUDA安装路径中这三个文件夹的父文件夹下面,Windows中默认安装路径为:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
win10安装cuda和cudnn

上一篇:RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED解决办法


下一篇:Linux中Tensorflow与CUDA和cuDNN的对应关系表