通俗的讲解深度学习中CUDA,cudatookit,cudnn和pytorch的关系
作者：江江ahh 发布时间：2023-05-02 05:07:34
CUDA
CUDA是显卡厂商NVIDIA推出的运算平台。CUDA™是一种由NVIDIA推出的通用并行计算架构，是一种并行计算平台和编程模型，该架构使GPU能够解决复杂的计算问题。CUDA英文全称是Compute Unified Device Architecture。
我们可以通过nvidia-smi命令查看cuda版本号。
如上图，我的电脑cuda版本是11.7
CUDA Toolkit
CUDA Toolkit可以理解成一个工具包，主要包含了CUDA-C和CUDA-C++编译器、一些科学库和实用程序库、CUDA和library API的代码示例、和一些CUDA开发工具。
cudatookit版本有时会被简称为cuda版本，这也是我们经常搞混的一个原因。
cuDNN
cuDNN的全称为NVIDIA CUDA® Deep Neural Network library，是NVIDIA专门针对深度神经网络中的基础操作而设计基于GPU的加速库。cuDNN为深度神经网络中的标准流程提供了高度优化的实现方式。
简单来说，cuDNN是基于CUDA的深度学习GPU加速库，有了它才能在GPU上完成深度学习的计算。
Pytorch
pytorch是基于CUDA的深度学习框架，因此，pytorch的版本必须依赖于cuda toolkit的版本。
如果你在读上面的一些名词的时候感觉模糊不清，那么可以直接来看下面的总结。（当然还是建议把不懂的地方搜索明白）
CUDA看作是一个工作台，上面配有很多工具，如锤子、螺丝刀等。cuDNN是基于CUDA的深度学习GPU加速库，有了它才能在GPU上完成深度学习的计算。它就相当于工作的工具，比如它就是个扳手。但是CUDA这个工作台买来的时候，并没有送扳手。想要在CUDA上运行深度神经网络，就要安装cuDNN，就像你想要拧个螺帽就要把扳手买回来。这样才能使GPU进行深度神经网络的工作，工作速度相较CPU快很多。
总结一下，食物链顶端的是CUDA这个工作台，也就是我们电脑的硬件配置，我们要先查看它的版本，然后根据这个来看我们都能下载多高的cudatookit版本，然后根据cudatookit版本来选择cudnn版本和可支持的pytorch版本
因此配置环境的流程为：
查看CUDA版本 ——> 选择cudatookit版本 ——> 选择cudnn版本 + pytorch版本
来源：https://blog.csdn.net/m0_68239345/article/details/128803487
0
投稿猜你喜欢
前言本文主要给大家介绍了关于Django快速分页的相关内容，分享出来供大家参考学习，下面话不多说了，来一起看看详细的介绍吧。分页在web开发- 一个正则表达式就是由普通字符（例如字符 a 到 z）以及特殊字符（称为元字符）组成的文字模式。该模式描述在查找文字主体时待匹配的一个或多个字
- 守来说，AJAX在现在是热得不能再热的技术。没有人能否认，它拥有大批的支持者。在CNN上，它从二月份的一个不被看好的词语到十月份成长成一个初
- 前言最近又在重新学习编译原理了，其实两年前也复习过，当初是为了能实现通过 MySQL 的 DDL 生
- 抽象工厂模式抽象工厂模式是一种创建型设计模式， 它能创建一系列相关的对象， 而无需指定其具体类。抽象工厂定义了用于创建不同产品的接口， 但将
- 日历功能在个人博客网站及一些任务类网站都有广泛的应用，当然，在一些通用网站的后台也不乏这些日历功能的综合应用，所以，一个结构合理、代码清晰的
- 插入视图的条件: 1.如果视图是基于一个基础表产生的,那么这就称为非连接视图,所有的非连接视图都是可以更新的,也就是说可以在该视图上进行,I
- 本篇介绍Python字典的常见操作。修改字典元素，如图。添加字典元素，如图。删除字典元素del方法，如图。删除字典元素clear方法，如图。
- 比如有两个模块，一个aa.py，一个bb.py 代码如下：aa.py:#encoding:utf-8import bba=1bb.py:#e
- asp fso type属性取得文件类型代码是用来返回类型指定的文件或文件夹。语法FileObject.Type FolderObject.
- 在昨天的文章，《 block 和 inline 的区别是？》里，我给大家留了个问题——LI 元素到底是block level 的，还是 in
- 写在前面从本节开始，计算机视觉教程进入第三章节——图像特征提取。在本章，你会见到一张简简单单的图片中
- 本文主要研究的是Python编程turtle的实例，绘制一个七彩蟒蛇。。具体如下。第2周的课后练习里，有一道题目，要求修改“蟒蛇绘制”程序，
- 前言要想学好爬虫，必须把基础打扎实，之前发布了两篇文章，分别是使用XPATH和requests爬取网页，今天的文章是学习Beautiful
- 1 . 如何让自己的本地APACHE服务器支持.htaccess 如何让自己的本地APACHE服务器支持”.htaccess”呢?其实只要简
- 在软件项目实施的时候，数据导入一直是项目人员比较头疼的问题。其实，在SQL Server中集成了很多成批导入数据的方法。有些项目实施顾问头疼
- <?php echo CHtml::link('错误链接','user/register')?>
- 前言最新需要做一个小工具，让协作部门能够获取到服务器上的一些资源讯息，因为工具是pyqt写的所以牵扯到用python链接linux的问题，这
- 原文地址：30 Days of Mootools 1.2 Tutorials - Day 19 - TooltipsMootools 1.2
- MySQL服务器有几个影响其操作的参数（变量）。如果缺省的参数值不合适，可以将其修改为对服务器运行环境更合适的值。例如，如果您有大量的内存，
