趣百科

cuDNN 安装教程

编辑:Simone 2024-11-26 18:40:39 601 阅读

cuDNN 安装教程

本人学生党一枚,最近入了深度学习的坑。网上关于cuDNN安装教程不胜枚举,个人在安装cuDNN过程中填了不少坑。这篇经验是我上一篇关于Win10系统下安装tensorflow-gpu版的补充。将自己的经验总结一下,希望能帮助到像我一样的童鞋。

cuDNN是用于深度神经网络的训练的加速库。从官网上下载的cuDNN是一个压缩包,解压缩以后对应三个文件夹。我们的目的是将这三个文件夹中的文件对应的放到CUDA的文件目录下。

找到你的CUDA文件安装目录。下图是我的CUDA文件安装目录,如果你当时是默认的安装路径,那基本上应该和我的路径是一致的。你可以根据红线标注的路径对应查找。

我们发现,两个文件目录下都存在“bin”"include0""lib"三个文件夹,你需要将cuDNN中的文件夹下的文件对应的拷贝到CUDA的文件夹下,作为原有文件的补充,说的通俗一点,就是将cuDNN文件夹拆分。

下面添加环境变量。打开环境变量面板(打开方法就不详细说了,你可以用“Cortana”直接搜索,也可以在系统设置里找到),编辑系统变量下的 “Path”变量。此处需要添加两个变量:1.C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin 2. C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\CUPTI\libx64 具体如下图所示:

安装到这里,就大功告成了!!

版权声明:本站【趣百科】文章素材来源于网络或者用户投稿,未经许可不得用于商用,如转载保留本文链接:https://www.qubaik.com/article/82609.html

相关推荐