首页 > 其他分享 >pytorch安装: cuda、cudatoolkit、torch版本对照

pytorch安装: cuda、cudatoolkit、torch版本对照

时间:2024-09-12 13:35:29浏览次数:12  
标签:torch cudatoolkit pytorch CUDA 版本 显卡 安装

PyTorch 官网上有如下安装对照表,同时也有历史版本安装对照表

image.png

从零开始配置python深度学习环境大概有如下配置步骤:

方案一: 电脑安装显卡驱动,然后安装CUDAcuDNN,安装miniconda3。前面都是在电脑基础环境配置,后面的操作都是在conda环境中,安装torchcudatoolkits等深度学习包
方案二:电脑安装显卡驱动,后直接安装miniconda,后续步骤在conda环境中进行操作,安装torchcudatoolkits等步骤

两种方案的区别,后文会讲解:


1. 配置显卡驱动(必须)

2.基础环境:CUDA,cuDNN(可选)

3.虚拟环境:配置 cudatoolkit(必须)

显卡驱动安装

显卡驱动下载:GeForce® 驱动程序,在此页面找到自己nvidia显卡对应的驱动程序。
image.png

宿主机CUDA 配置

如果本机配备独立显卡,且安装显卡驱动后,可以在终端运行 nvidia-smi 检查显卡的支持的最高CUDA 版本,我手头有不同架构的设备,以下列出几种查询方案:

自己 4060 win电脑

image.png

黄框表示能够支持的最高 CUDA 版本,即安装的 CUDA 版本不能高于此数字。

Jeston Xavier NX 开发板
开发板属于 Aarch64 架构,与个人 PC 上常见的 X86 架构不同,需要执行如下命令才能查看:

cd /usr/local/cuda/samples/1_Utilities//deviceQuery/
sudo make
./deviceQuery 

image.png

由于 Jeston 是显存和物理内存共用,不能通过 nvidia-smi 查看显卡信息

如果要查看 window 电脑是否安装了 CUDA,可以执行如下步骤

nvcc --version
# 可以看到如下输出
> nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2021 NVIDIA Corporation Built on ... Cuda compilation tools, release 11.2, V11.2.152

cudatoolkit 版本

cudatoolkitcudatoolkit 是 NVIDIA CUDA 工具包的一个精简版本,专为在 Conda 环境中使用而设计,其为 python 环境中的 GPU 加速计算提供必要的组件。适用于 PyTorch、TensorFlow 等框架。

显卡的 CUDA 版本:这是指通过显卡驱动安装的 CUDA 版本。可以通过 nvidia-smi 命令查看系统中当前安装的 CUDA 版本。

CUDA 版本兼容性:

  • 较新的 CUDA 版本通常向后兼容旧 GPU
  • 但新 GPU(如 Ampere 架构)需要较新的 CUDA 版本才能充分发挥性能

cudatoolkit其与系统CUDA版本的关系:

  • cudatoolkit可以与系统级CUDA共存
  • 通常使用系统级CUDA驱动,但运行时库来自cudatoolkit
# 指定版本安装
$ conda install cudatoolkit=11.2
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

该命令安装了指定版本的cudatoolkit, cudatoolkit 是一个已编译好的 CUDA 库,它会在运行时被 PyTorch 使用,而不依赖于系统全局的 CUDA 安装。同时 torch 也会自动安装与指定版本的 PyTorch 兼容的 cuDNN

此链接为 pytorch 和 cudatoolkit 版本对应关系:pytorch各版本对照

torchtorchvisionpython 对照表

torch torchvision Python
main / nightly main / nightly >=3.8, <=3.12
2.3 0.18 >=3.8, <=3.12
2.2 0.17 >=3.8, <=3.11
2.1 0.16 >=3.8, <=3.11
2.0 0.15 >=3.8, <=3.11
1.13 0.14 >=3.7.2, <=3.10
1.12 0.13 >=3.7, <=3.10
1.11 0.12 >=3.7, <=3.10
1.10 0.11 >=3.6, <=3.9
1.9 0.10 >=3.6, <=3.9
1.8 0.9 >=3.6, <=3.9
1.7 0.8 >=3.6, <=3.9
1.6 0.7 >=3.6, <=3.8
1.5 0.6 >=3.5, <=3.8
1.4 0.5 ==2.7, >=3.5, <=3.8
1.3 0.4.2 0.4.3 ==2.7, >=3.5, <=3.7
1.2 0.4.1 ==2.7, >=3.5, <=3.7
1.1 0.3 ==2.7, >=3.5, <=3.7
<=1.0 0.2 ==2.7, >=3.5, <=3.7

image.png

标签:torch,cudatoolkit,pytorch,CUDA,版本,显卡,安装
From: https://www.cnblogs.com/jacexu016/p/18409959

相关文章

  • 【小白深度教程 1.16】手把手教你使用 Pytorch3D(1)使用 3D 损失函数来拟合 Mesh
    【小白深度教程1.16】手把手教你使用Pytorch3D(1)使用3D损失函数来拟合Mesh1.安装和导入模块2.加载.obj文件并创建Mesh对象3.可视化源Mesh和目标Mesh4.迭代优化进行拟合5.可视化损失6.保存结果在这篇文章中,我们将学习如何使用3D损失函数变形源......
  • 基于深度学习的骨龄检测识别系统(PyTorch+Pyside6+YOLOv5模型)
    骨龄是骨骼年龄的简称,需要借助于骨骼在X光摄像中的特定图像来确定。通常要拍摄左手手腕部位的X光片,医生通过X光片观察来确定骨龄。在2006年,《中华-05》骨龄标准被编入《中华人民共和国行业标准》,成为中国目前唯一的行业骨龄标准。而在《中华-05》中作者一般推荐使用RUS-CHN计......
  • 使用PyTorch Lightning力量精简空间分析
    大家好,随着人工智能热潮的全面兴起,PyTorchLightning库正在获得越来越多的关注。其特别突出的地方在于简化复杂的机器学习操作,即使对于非开发者也是如此。深度学习和部分机器学习中的许多挑战性方面,如多GPU训练和实验跟踪,都由该框架自动处理,同时保持了PyTorch的灵活性和高效性......
  • 每天五分钟玩转深度学习框架PyTorch:获取神经网络模型的参数
    本文重点当我们定义好神经网络之后,这个网络是由多个网络层构成的,每层都有参数,我们如何才能获取到这些参数呢?我们将再下面介绍几个方法来获取神经网络的模型参数,此文我们是为了学习第6步(优化器)。获取所有参数Parametersfromtorchimportnnnet=nn.Sequential(nn.Linear(4......
  • 大模型书籍推荐:《Deep Learning with PyTorch》PyTorch深度学习实战,从核心理论到实战!(
    一、PyTorch深度学习实战PyTorch核心开发者教你使用PyTorch创建神经网络和深度学习系统的实用指南。这本书详细讲解整个深度学习管道的关键实践,包括PyTorch张量API、用Python加载数据、监控训练以及对结果进行可视化。PyTorch核心知识+真实、完整的案例项目,快速提升读者动手能......
  • 【pytorch(cuda)】基于DQN算法的无人机三维城市空间航线规划(Python代码实现)
       ......
  • torch中 nn.BatchNorm1d
    nn.BatchNorm1d是PyTorch中的一个用于一维数据(例如序列或时间序列)的批标准化(BatchNormalization)层。批标准化是一种常用的神经网络正则化技术,旨在加速训练过程并提高模型的收敛性和稳定性。它通过对每个输入小批次的特征进行归一化处理来规范化输入数据的分布。在一维数据上使......
  • torch.bmm释义
    torch.bmm是PyTorch中的一个函数,用于执行批量矩阵相乘(batchmatrixmultiplication)的操作。它用于计算两个具有相同批次大小的三维张量的矩阵乘法。在矩阵乘法中,两个矩阵的维度必须满足一定的条件。对于torch.bmm函数,它要求输入的两个张量都具有三个维度,形状分别为(batch_siz......
  • liveportrait_pytorch可以实现静态图模仿动态图面部动作AIGC模型
    LivePortrait论文LivePortrait:EfficientPortraitAnimationwithStitchingandRetargetingControlhttps://arxiv.org/pdf/2407.03168模型结构模型基于facevid2vid,并在此基础上进行改进。主要为,使用ConvNeXt-V2-Tiny作为backbone将原始的规范隐式关键点检测器L、头......
  • 每天五分钟玩转深度学习框架PyTorch:将nn的神经网络层连接起来
    本文重点前面我们学习pytorch中已经封装好的神经网络层,有全连接层,激活层,卷积层等等,我们可以直接使用。如代码所示我们直接使用了两个nn.Linear(),这两个linear之间并没有组合在一起,所以forward的之后,分别调用了,在实际使用中我们常常将几个神经层组合在一起,这样不仅操作方便,而且......