显卡驱动CUDA 和 pytorch CUDA 之间的区别

更新时间：2023年03月16日 08:43:38 作者：null_one

本文主要介绍了显卡驱动CUDA 和 pytorch CUDA 之间的区别，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧

写在前面

我的 CUDA 版本是什么？这个问题本身就是有问题的，因为没有搞清楚cuda的分类

这里的 CUDA 说的是 Driver CUDA 还是 Runtime CUDA？
如果是 Runtime CUDA，那系统可能有多个，说的是哪个？用 apt-get 装的？PyTorch 自带的？

想知道答案的话往下看，相信看完之后应该有所感悟

如何查看CUDA 版本

常见的查看CUDA版本的指令有如下几个

nvidia-smi
nvcc --version
python -c "import torch; print(torch.version.cuda)"

nvidia-smi

在这里插入图片描述

nvcc --version

我先在用的是Windows电脑，这里是一张Linux 网图

在这里插入图片描述

python -c “import torch; print(torch.version.cuda)”

在这里插入图片描述

为何三个指令三种结果？

nvidia-smi

这个命令一旦装完显卡驱动（nvidia-driver）就可以使用了，驱动程序提供最底层的程序接口，直接与硬件打交道。插上显卡的计算机也必须要装上驱动程序才能使用显卡。所以 nvidia-smi 的输出几乎都是显卡的硬件信息。
系统同一时间只能安装一个版本的驱动程序，如果安装多套版本可能会导致未定义的错误，另外驱动程序保持最新版本即可。

那这里的 CUDA Version 是代表什么呢？其实驱动程序也有 CUDA，叫做 Driver CUDA，这里指的是 Driver CUDA 的版本。而几乎所有的框架程序，所使用的 CUDA 都不是 Driver CUDA，而是 NVIDIA 提供的更上层的编程接口，叫做 Runtime CUDA

nvcc 和 torch.version.cuda

首先 nvcc 是一个编译器，这个编译器是用户在安装 Runtime CUDA 的时候附带的。因此 nvcc 打印的是 Runtime CUDA 的版本。同样的，torch.version.cuda 打印的也是 Runtime CUDA 的版本，那这两个版本为什么也不同呢？我们需要先了解Runtime CUDA。

Runtime CUDA

Runtime CUDA 是 NVIDIA 封装的上层接口，几乎所有的应用程序，都是直接调用 Runtime CUDA 的 API，而 Runtime CUDA 内部调用 Driver 的接口。所以通常所说的 CUDA 都是指的 Runtime CUDA（除非是驱动开发的人员）。另外 NVIDIA 有一个规则，Runtime CUDA 的版本号必须小于等于Driver CUDA 的版本号，所以 nvidia-smi 输出的 CUDA 版本可以理解为该系统所能安装的最高 CUDA 版本号是多少。

安装方法

nvcc --version 输出的是当前安装的 CUDA 的版本，而系统同一时间还可以安装多套 CUDA，并且安装的方式还多种多样，一般来说有以下几种安装方式：

PyTorch 本身自带 CUDA

# 会自动安装 CUDA 10.2
pip3 install torch

# 会自动安装 CUDA 11.3
pip3 install torch --extra-index-url https://download.pytorch.org/whl/cu113

# 会自动安装 CUDA 11.6
pip3 install torch --extra-index-url https://download.pytorch.org/whl/cu116

有人可能会问：为什么装了 PyTorch 之后，没有 nvcc 这个命令，或是 nvcc --version 显示的 CUDA 和 PyTorch 安装的 CUDA 版本对不上？

上文说了，一个系统可能有多套 CUDA，如果 nvcc --version 显示的 CUDA 对不上号，是因为系统本身安装了其他版本的 CUDA。另外，PyTorch 自带的 CUDA 只包含库文件，没有 nvcc 编译器，所以没有 nvcc 这个命令。

通过 Ubuntu 的官方源安装

sudo apt-get install nvidia-cuda-toolkit

通过 conda 安装

conda search -c conda-forge cudatoolkit  # 列出所有可以安装的版本
conda install -c conda-forge cudatoolkit=xx.x.x

通过官方安装包访问下面的链接，可以下载到不同版本的安装包：toolkits

torch官网有详细的安装cuda的步骤，自己去官网找一下就好了

到此这篇关于显卡驱动CUDA 和 pytorch CUDA 之间的区别的文章就介绍到这了,更多相关显卡驱动CUDA 和 pytorch CUDA内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

pytest框架之fixture详细使用详解
这篇文章主要介绍了pytest框架之fixture详细使用详解，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2021-04-04
利用Python进行时间序列数据分析与可视化的代码示例
随着时间序列数据在金融、气象、生态等领域的广泛应用,利用Python进行时间序列数据分析和可视化已成为重要的技能之一,本文将介绍如何使用Python进行时间序列数据分析和可视化,并给出相应的代码示例,需要的朋友可以参考下
2023-11-11
Python实现微信高效自动化操作
在如今数字化时代,人们对于效率的追求越来越强烈,而PyAutoGUI和Pyperclip作为Python中的两个强大库,为我们实现自动化操作提供了便利,下面我们就来看看如何利用这两个库实现微信自动化操作吧
2023-10-10
termux中matplotlib无法显示中文问题的解决方法
这篇文章主要介绍了termux中matplotlib无法显示中文问题的解决方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2021-01-01
python中的selenium安装的步骤（浏览器自动化测试框架）
这篇文章主要介绍了python中的selenium安装的步骤，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-03-03
通过Python的jieba库对文本进行分词
Python的jieba库是一个中文分词工具，它可以将一段中文文本分割成一个一个的词语，方便后续的自然语言处理任务，如文本分类、情感分析等，本文给大家介绍如何通过Python的jieba库对文本进行分词，文中详细的代码示例,需要的朋友可以参考下
2023-05-05
python爬虫可以爬什么
在本篇文章里小编给大家整理的是关于python爬虫的作用地方以及相关知识点，需要的朋友们可以学习下。
2020-06-06
Python中用户输入与while循环详情
这篇文章主要介绍了Python中用户输入与while循环详情，，包括如何接收用户输入并进行处理，在程序满足一定的条件时让程序一直运行，通过获取用户输入并学会控制程序在用户想要结束时退出循环，即可编写出交互式程序，下文详细内容介绍,需要的朋友可以参考一下
2022-03-03
浅析python内置模块collections
collections是Python内建的一个集合模块，提供了许多有用的集合类。这篇文章主要介绍了python内置模块collections的相关知识，非常不错，具有一定的参考借鉴价值,需要的朋友可以参考下
2019-11-11
解决Tensorflow使用pip安装后没有model目录的问题
今天小编就为大家分享一篇解决Tensorflow使用pip安装后没有model目录的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2018-06-06