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

news/2024/9/17 17:46:43/文章来源:https://www.cnblogs.com/jacexu016/p/18409959

在 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

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.hqwc.cn/news/795998.html

如若内容造成侵权/违法违规/事实不符,请联系编程知识网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

AI实战 | 领克汽车线上营销助手:全面功能展示与效果分析

本篇文章的主要目的是为大家提供实现思路,以及如何更好地开发一个助手,而不仅仅是简单地进行拆解。如果采取拆解的方式,一篇文章可能会长达2万+字,还需要配以数十张图片,这将会非常繁琐。因此,针对拆解的详细内容,我计划单独制作一期视频,以帮助大家更清晰地理解。感谢…

ThreadLocal源码分析-

ThreadLocal源码分析 ThreadLocal是解决线程安全问题的一种方法,它通过为每个线程提供一个独立的变量副本避免了变量并发访问的冲突问题。一个ThreadLocal变量只与当前自身线程相关,对其他线程是隔离的。下面这段代码展示了ThreadLocal的使用。 public class test {private s…

Python打包完成后报错,如何解决?

大家好,我是Python进阶者。 一、前言 前几天在Python最强王者交流群【钟爱一生】问了一个Python打包处理数据的问题,问题如下: 打包完成后报错:发生错误:Missing optional dependency openpyxl.Use pip or conda to install openpyxl.哪位大佬帮我看一下错在哪了?二、实现…

408存储系统大题笔记

咸鱼25计组强化课P2的笔记,有点简陋。 课后需要把第三章的课后大题(真题做一下) Cache类题目做题要注意的点!!PA的位数=Cache地址总位数 Cache总行数 Cache块大小 第2和第3相乘得Cache数据区总大小(!=Cache总大小) 映射方式 一致性问题:写策略(直写/回写) 替换算法 CPU…

浅谈 C# 中的顶级语句

前言 在C# 9版本中引入了一项新特性:顶级语句,这一特性允许在不显式定义 Main 方法的情况下直接编写代码。 传统的写法namespace TestStatements{ internal class Program { static void Main(string[] args) { foreach (var arg in args) …

清理C盘缓存,清理C盘缓存的详细方法步骤

清理C盘缓存是优化Windows系统性能的重要步骤之一。以下是一些详细的方法来清理C盘缓存: 一、使用Windows内置工具磁盘清理 打开磁盘清理: 通过文件资源管理器右键点击C盘,选择“属性”,然后点击“磁盘清理”。 或者在“设置”中选择“系统” > “存储”,查看C盘的存储…

idea中的docker部署配置

注意:确认本地已安装docker环境 第一步:idea安装docker插件:设置-插件-docker第二步:配置Dockerfile文件FROM harbor.chint.com/wz-build-env-public/openjdk:17 AS base# 项目的端口,内部服务端口 EXPOSE 8808# 切换到容器内部的 /workdir目录 WORKDIR /workdir# 添加要运…

第十一章 图论 Part8

dijkstra(朴素版) 拓扑排序目录单源最短路径算法dijkstra(朴素版)适用范围(权值不能为负数的单源最短路径)思路算法正确性拓扑排序思路心得 单源最短路径算法 dijkstra(朴素版) 适用范围(权值不能为负数的单源最短路径) 思路 基本类似Prim算法,只是新加入(确定)点…

Dockerfile 实战指南:轻松掌握容器化部署!

Dockerfile 非常重要,在实际工作中,使用 Docker 绝不是敲敲一些常用命令即可。Dockerfile 几乎贯穿微服务的全部内容,务必掌握。Dockerfile 非常重要,在实际工作中,使用 Docker 绝不是敲敲一些常用命令即可。Dockerfile 几乎贯穿微服务的全部内容,务必掌握。 不要求能从头…

Hodgkin-Huxley Model 完全推导

Hodgkin-Huxley Model 膜电流计算的相关公式的完全推导,从物理、数学到公式推导。Ciallo~(∠・ω< )⌒★ 我是赤川鹤鸣。本文假设您已经初步了解了 Hodgkin-Huxley Model,这里只是针对其中的公式的一些推导。不会对其优缺点、特性、应用等进行详述。物理基础知识如果已学…

PbootCMS自定义前台404错误页面

PbootCMS已经内置支持自定义内容地址错误情况下错误页面的自定义功能,只需要在站点根目录下定义404.html文件即可扫码添加技术【解决问题】专注中小企业网站建设、网站安全12年。熟悉各种CMS,精通PHP+MYSQL、HTML5、CSS3、Javascript等。承接:企业仿站、网站修改、网站改版、…

Nuxt Kit 的使用指南:从加载到构建

title: Nuxt Kit 的使用指南:从加载到构建 date: 2024/9/12 updated: 2024/9/12 author: cmdragon excerpt: 摘要:本文详细介绍了Nuxt Kit的使用方法,包括如何使用loadNuxt加载配置、buildNuxt进行项目构建、loadNuxtConfig单独加载配置以及writeTypes生成TypeScript配置,…