通俗的讲解深度学习中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
投稿猜你喜欢
- 如何用WSH获取机器的IP配置信息？我们用VBSCRIPT转换了： Option Explicit Dim&n
- 这个是今年年初写的一篇，拿出来温习下。指针让程序结构变得混乱，也让程序执行效率提高，因此在oo的语言中不提倡指针的使用，使得程序结构清晰易读
- 说明C# 调用 Python 程序有多种方式，本篇用的是第 4 种：nuget的ironPython；用 c/c++ 调用python，再封
问题描述最近～ 发现对series里的元素操作挺复杂的，用for loop + Series.iloc[i]会发生卡死的状况，那么，lamb- 不知道大家在做页面的时候会不会遇到样式定义不生效的问题，基本的表现就是怎么改样式都没显示或只有某些浏览器正常，这时通常需要做下面的几步：确认
- 本文实例讲述了js找出5个数中最大的一个数和倒数第二大的数实现方法。分享给大家供大家参考，具体如下：最大的一个数 ：方法1：<scri
- 前言：平时工作没有养成分类的习惯，整个桌面杂乱无章都是文档和资料，几乎快占满整个屏幕了。所以必须要整理一下了，今天我们来看下用python如
- 内容摘要： Request和Response这两个对象是ASP所提供的内置对象中最常用的两个。在浏览器（或其他用户代理）和Web服
- Server对象主要是给编程人员提供一些方便的对象和属性。（1）ScriptTimeout属性：<％Server.ScriptTime
- 目录前言cv2.drawMarker()函数说明参数说明利用鼠标回调函数交互式画点例1，简单的例子例2，删除功能总结前言这里所谓画点的意思是
- 一、Matplotlib 绘图在数据分析中，数据可视化也非常重要，通过直观的展示过程、结果数据，可以帮助我们清晰的理解数据，进而更好的进行分
- 学习Django框架时,创建一个引擎及索引时报错,具体报错如下:执行命令:python3 manage.py rebuild_index报如
- 前几天翻出以前写的一个纯CSS仿微软经典菜单，现在看来才感叹，微软的经典菜单确实很经典，至少看起来不觉得厌烦。感叹归感叹，想想既然可以实现下
- 1. Python的数据类型上一遍博文已经详细地介绍了Python的数据类型，详见链接Python的变量命名及数据类型。在这里总结一下Pyt
- 前言自从学过存储过程后，就再也没有碰过存储过程，这是毕业后写的第一个存储过程。因为项目里设备的种类比较多，分别存在不同的数据表中，java中
- 首先是创建一个类，继承于ActionResult，记住要引用System.Web.Mvc命名空间，如下： public class Imag
- 前言在8之前的版本中，对于全局变量的修改，其只会影响其内存值，而不会持久化到配置文件中。数据库重启，又会恢复成修改前的值。从8开始，可通过S
- wed的打印方法具我自己懂得知道的有： 1、JQuery插件Jqprint实现 2、JQery打印插件PrintArea实现网页打印 3、C
- 1005：创建表失败1006：创建数据库失败1007：数据库已存在，创建数据库失败1008：数据库不存在，删除数据库失败1009：不能删除数
- 英文原文：http://www.456bereastreet.com/archive/200601/css_3_selectors_expl
