关于cuda、cudnn环境配置

cuda、cudnn环境配置

一、cuda、cudnn概念及关系

1、什么是cuda/h3>

CUDA(ComputeUnified Device Architecture),是显卡厂商NVIDIA推出的运算平台。 CUDA是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题

2、什么是cudnn/h3>

NVIDIA cuDNN是用于深度神经网络的GPU加速库。它强调性能、易用性和低内存开销。NVIDIA cuDNN可以集成到更高级别的机器学习框架中,如谷歌的Tensorflow、加州大学伯克利分校的流行caffe软件。简单的插入式设计可以让开发人员专注于设计和实现神经网络模型,而不是简单调整性能,同时还可以在GPU上实现高性能现代并行计算。

3、cuda和cudnn关系/h3>

CUDA看作是一个工作台,上面配有很多工具,如锤子、螺丝刀等。cuDNN是基于CUDA的深度学习GPU加速库,有了它才能在GPU上完成深度学习的计算。它就相当于工作的工具,比如它就是个扳手。但是CUDA这个工作台买来的时候,并没有送扳手。想要在CUDA上运行深度神经网络,就要安装cuDNN,就像你想要拧个螺帽就要把扳手买回来。这样才能使GPU进行深度神经网络的工作,工作速度相较CPU快很多。

4、CUDNN不会对CUDA造成影响/h3>

从官方安装指南可以看出,只要把cuDNN文件复制到CUDA的对应文件夹里就可以,即是所谓插入式设计,把cuDNN数据库添加CUDA里,cuDNN是CUDA的扩展计算库,不会对CUDA造成其他影响。

二、查看当前cuda版本

1、安装过cuda的

方法一:win+R 输入nvidia-smi查看

在cmd命令行中输入nvidia-smi查看NVIDIA显卡driver version以及可安装的最高CUDA版本,如下图本电脑所示,driver version为511.65,可安装的最高CUDA版本为11.6

关于cuda、cudnn环境配置
(2)选择左下角的系统信息,进入
关于cuda、cudnn环境配置

方法三:win+R 输入cmd命令nvcc -V

关于cuda、cudnn环境配置
以上四种方法都是可以查看cuda版本的,我做了个总结。

2、没有安装过cuda的

(1)下载

在NVIDIA官网下载对应版本的显卡驱动:显卡驱动下载地址
进入官网,选择相应的版本和系统进行下载

关于cuda、cudnn环境配置

(2)安装

a.右击下载好的cuda_10.2.89_441.22_win10.exe软件,以管理员身份运行,如下图所示:

关于cuda、cudnn环境配置
关于cuda、cudnn环境配置
d.这里我们选择【自定义】,然后下一步。
关于cuda、cudnn环境配置
f.直接默认装在C盘,这样可以防止各种出错。
关于cuda、cudnn环境配置
点击next等待安装即可

注:因为我自己已经安装了,所以上图使用的是别人的,安装步骤都是一样的,感谢博主的图片,链接已放在参考资料里面。

三、查看当前cudnn版本

1、安装过cudnn的

进入cuDNN安装目录下查看文件 cudnn.h
C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0includecudnn.h
cudnn的版本为7.6.1

关于cuda、cudnn环境配置
注:一点要选择CUDA版本对应的cuDNN下载!!!
一点要选择CUDA版本对应的cuDNN下载!!!
一点要选择CUDA版本对应的cuDNN下载!!!

重要的事情说三遍!!

安装cudnn

下载好cudnn压缩包之后进行解压(具体版本大家自行对应,我这里只是举例,版本不对应)

关于cuda、cudnn环境配置
把压缩包里面的内容复制到C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.0里面就可以了
关于cuda、cudnn环境配置
则说明CUDA和CUDNN已经配置成功!

参考链接:https://www.jianshu.com/p/622f47f94784
http://t.csdn.cn/99GKZ
http://t.csdn.cn/d08j2

文章知识点与官方知识档案匹配,可进一步学习相关知识CUDA入门技能树首页概览1588 人正在系统学习中

来源:翠小白

声明:本站部分文章及图片转载于互联网,内容版权归原作者所有,如本站任何资料有侵权请您尽早请联系jinwei@zod.com.cn进行处理,非常感谢!

上一篇 2022年10月8日
下一篇 2022年10月8日

相关推荐