cuDNN 安装教程
编辑:Simone
2024-11-26 18:40:39
601 阅读
本人学生党一枚,最近入了深度学习的坑。网上关于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