【超详细教程】2025年3月最新Pytorch安装教程(同时讲解安装CPU和GPU版本)

news2025/4/2 22:01:26

目录

  • 一、前言
  • 二、pytorch简介
  • 三、安装准备工作
    • 3.1、下载Anaconda
  • 四、判断是否有NVIDIA显卡
  • 五、安装pytorch-CPU版本
  • 六、安装pytorch-GPU版本
    • 6.1、查看CUDA显卡驱动版本
    • 6.2、安装CUDA
    • 6.3、安装CuDNN(加速器)
    • 6.4、安装pytorch-GPU
    • 6.5 其他方法安装
      • 注意
  • 七、pytorch-GPU安装验证

一、前言

在深度学习和人工智能领域,PyTorch已成为最受欢迎的开源框架之一。随着技术的不断进步,PyTorch也在持续更新和优化。本文将为您带来2025年最新版本的PyTorch安装教程,并详细解析如何安装支持CPU和GPU的版本。通过本文,您将轻松掌握PyTorch的安装过程,为您在深度学习的探索之旅中提供强大的工具支持。
在这里插入图片描述

二、pytorch简介

Pytorch是基于torch的python版本的机器学习库,是由Facebook开源的神经网络框架,专门针对 GPU 加速的深度神经网络(DNN)编程。 Torch 是一个经典的对多维矩阵数据进行操作的张量(tensor )库,在机器学习和其他数学密集型应用有广泛应用。与Tensorflow的静态计算图不同,pytorch的计算图是动态的,可以根据计算需要实时改变计算图。但由于Torch语言采用 Lua,导致在国内一直很小众,并逐渐被支持 Python 的 Tensorflow 抢走用户。作为经典机器学习库 Torch 的端口,PyTorch 为 Python 语言使用者提供了舒适的写代码选择。

在这里插入图片描述

三、安装准备工作

在开始安装PyTorch之前,我们需要进行一些准备工作。首先,确保您的计算机上已安装了合适版本的Python。PyTorch需要Python 3.6或更高版本。同时,您还需要安装pip包管理器,用于安装PyTorch和其他依赖项。所以我们可以通过安装Anaconda来创建一个适合深度学习的Python环境。以下是详细的安装步骤。

3.1、下载Anaconda

首先,访问Anaconda官网(Free Download | Anaconda),选择适合你操作系统的版本,并下载安装包,详细的安装步骤可参考我之前的文章:Anaconda在Windows上的安装与使用
在这里插入图片描述

在这里插入图片描述

四、判断是否有NVIDIA显卡

在安装PyTorch之前,我们需要判断您的计算机是否安装了NVIDIA显卡,因为PyTorch的GPU版本需要NVIDIA显卡来加速计算。您可以通过以下步骤来判断:

① 打开设备管理器:
在Windows上,按下Win键和X键,然后选择“设备管理器”。
在macOS上,打开“系统偏好设置”,选择“硬件”选项卡,然后点击“设备管理器”。

② 查看显示适配器:在设备管理器中,展开“显示适配器”或“图形处理器”部分,查看是否有NVIDIA显卡的列表。如果有NVIDIA显卡,那么您的计算机适合安装PyTorch的GPU版本。

如果没有NVIDIA显卡,您可以只安装PyTorch的CPU版本。如果您有NVIDIA显卡,您还需要进行其他步骤来安装PyTorch的GPU版本。
在这里插入图片描述
(Windows系统——有NVIDIA显卡示例)

五、安装pytorch-CPU版本

在安装PyTorch之前,为了管理不同项目的Python环境,通常建议创建一个虚拟环境。虚拟环境可以帮助您隔离不同项目的依赖项,避免不同项目之间的冲突。以下是创建虚拟环境的步骤。

① 打开终端:在Windows上,打开Anaconda Prompt;在macOS和Linux上,打开终端;
在这里插入图片描述

② 创建虚拟环境:运行以下命令来创建一个新的虚拟环境。您可以将<env_name>替换为您喜欢的环境名称,例如“my_env”。

conda create -n <env_name> python=<version>

例如,要创建一个名为myenv的虚拟环境,其中Python版本为3.10,可以运行:

conda create -n my_env python=3.10

③ 激活虚拟环境:运行以下命令来激活刚刚创建的虚拟环境。在Windows上,使用activate命令;在macOS和Linux上,使用source命令。

conda activate <env_name>

例如,要激活名为my_env的虚拟环境,可以运行:

conda activate my_env

# To deactivate an active environment, use
conda deactivate

激活虚拟环境后,您将看到虚拟环境的命令提示符前缀显示为(<env_name>)。这意味着您现在正在使用该虚拟环境的Python解释器。
在这里插入图片描述

④ 安装PyTorch(CPU):在激活的虚拟环境中,使用pip安装PyTorch。根据您的需求,选择安装CPU版本。
在这里插入图片描述

pip3 install torch torchvision torchaudio

官网安装可能很慢(网络不佳和必要的时候可以添加国内源,清华国内源:https://pypi.tuna.tsinghua.edu.cn/simple)

pip install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple

在这里插入图片描述

在这里插入图片描述

这样就是安装成功啦~🥰(CPU版本)

六、安装pytorch-GPU版本

6.1、查看CUDA显卡驱动版本

在cmd命令行终端输入nvidia-smi,可以查看到版本为12.5。
在这里插入图片描述

6.2、安装CUDA

① 为什么安装pytorch-GPU前一定要安装CUDA和CuDNN呢?
在安装PyTorch-GPU之前,需要先安装CUDA和cuDNN的原因如下:
1. CUDA是NVIDIA推出的并行计算平台和API模型,它使得显卡可以用于图像渲染和计算以外的
目的,例如通用并行计算。PyTorch通过CUDA可以充分利用GPU的计算能力,加速深度神经网络> 的学习和推理过程。
2. cuDNN是CUDA的扩展库,专门针对深度神经网络中的基础操作提供高度优化的实现方式,
例如卷积、池化、规范化以及激活层的前向和后向过 程。使用cuDNN可以大大提高深度学习模型在GPU上的运行效率。

因此,在安装PyTorch-GPU之前,需要先安装CUDA和cuDNN,以便能够充分利用GPU的计算能力,加速深度神经网络的学习和推理过程。如果不安装CUDA和cuDNN,PyTorch-GPU将无法正常工作。

从官网下载对应的CUDA版本,由于我的显卡版本为12.5,我只需要安装小于或者等于12.5.0都是可以的,我安装的是12.5.0,(目前建议安装11.8,https://download.pytorch.org/whl相对比较完善)。

官网地址:CUDA Toolkit Archive | NVIDIA Developer
在这里插入图片描述

在这里插入图片描述

下载好后,双击安装包进行安装
在安装过程中选择路径的时候,第一次选择的路径下的文件夹会消失。
此处为临时安装目录(默认就可),后期安装完成会自动删除。
在这里插入图片描述

选择“精简”模式,接下来一直点“下一步”就行啦~😉,精简模式会默认安装到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.5这个文件夹
在这里插入图片描述
(开始安装,等待后续安装完成)

查看是否安装成功,在命令行输入以下指令进行检查,出现以下类似的输出就证明安装成功。

nvcc  -V

在这里插入图片描述

安装遇到的问题:
安装后不知道cuda安装到哪个文件夹下:【安装完CUDA后文件夹消失解答】

6.3、安装CuDNN(加速器)

通过官网进行安装CuDNN,网址:CUDA Deep Neural Network (cuDNN) | NVIDIA Developer
在这里插入图片描述
下载相对应的压缩包文件
在这里插入图片描述

安装并解压完后,将这几个文件夹复制到CUDA安装路径下,就已经安装完成啦😎
在这里插入图片描述
复制到cuda文件夹C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.5下。
CUDA安装默认路径:
Windows:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA
Linux:/usr/local/cuda
在这里插入图片描述

6.4、安装pytorch-GPU

在安装PyTorch之前,为了管理不同项目的Python环境,通常建议创建一个虚拟环境。虚拟环境可以帮助您隔离不同项目的依赖项,避免不同项目之间的冲突。以下是创建虚拟环境的步骤。

① 打开终端:在Windows上,打开Anaconda Prompt;在macOS和Linux上,打开终端;
在这里插入图片描述

② 创建虚拟环境:运行以下命令来创建一个新的虚拟环境。您可以将<env_name>替换为您喜欢的环境名称,例如“my_env”。

conda create -n <env_name> python=<version>

例如,要创建一个名为my_env的虚拟环境,其中Python版本为3.10,可以运行:

conda create -n myenv python=3.10

③ 激活虚拟环境:运行以下命令来激活刚刚创建的虚拟环境。在Windows上,使用activate命令;在macOS和Linux上,使用source命令。

conda activate <env_name>

例如,要激活名为my_env的虚拟环境,可以运行:

conda activate my_env

激活虚拟环境后,您将看到虚拟环境的命令提示符前缀显示为(<env_name>)。这意味着您现在正在使用该虚拟环境的Python解释器。

④ 安装PyTorch(GPU):在激活的虚拟环境中,使用pip安装PyTorch。根据您的需求,选择安装GPU版本。例如:(网络不佳和必要的时候可以添加国内源)
官网链接:PyTorch
在这里插入图片描述
命令

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

官网安装可能很慢(网络不佳和必要的时候可以添加国内源)
下面是一些镜像地址
清华镜像:https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
北京外国语大学镜像: https://mirrors.bfsu.edu.cn/anaconda/pkgs/main
阿里巴巴镜像:http://mirrors.aliyun.com/anaconda/pkgs/main

在这里插入图片描述

这样就是安装成功啦~🥰(GPU版本)

6.5 其他方法安装

(1)安装固定版本号

pip install torch==2.6.0+cu118 torchvision==0.21.0+cu118 torchaudio==2.6.0+cu118 -f
https://download.pytorch.org/whl/torch_stable.html

(2)离线安装torch,由于torch比较大,安装缓慢,可下载后离线安装

  • 离线包下载地址:download.pytorch.org/whl/torch_stable.html
  • 安装方式
    pip install torch-2.0.1+cu118-cp310-cp310-win_amd64.whl
    

注意

1)Pytorch与torchvision版本对应问题

Pytorch与torchvision版本配套

在这里插入图片描述

七、pytorch-GPU安装验证

注意:torch.cuda.is_available() 为True则GPU可用,False表示不可用

import torch
 
print(torch.__version__)
print(torch.version.cuda)
print(torch.cuda.is_available())  #输出为True,则安装成功

运行输出
.在这里插入图片描述

这样就是安装成功啦~🥰(GPU版本)

八、总结
🎉 恭喜您!您已经成功完成了PyTorch的安装教程。通过阅读本文,您了解了安装PyTorch的详细步骤,包括准备工作、判断显卡、创建虚拟环境、安装pytorch以及验证安装是否成功。 🚀

参考文章:
从零开始安装pytorch(包含GPU和CPU版本的安装)
【超详细教程】2024最新Pytorch安装教程(同时讲解安装CPU和GPU版本)

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

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

虚拟电商-话费充值业务(二)话费充值对接供应商模块开发

一、对接供应商模块开发 供应商对接模块chongba_recharge_supplier主要负责的就是调用外部的供应商系统进行充值下单&#xff0c;这种调用是一种基于HTTP协议的调用。 此外在供应商对接模块中主要是实现的业务逻辑有&#xff1a; 1&#xff1a;余额或押金不足情况下的失败轮…

c#winform,倒鸭子字幕效果,typemonkey字幕效果,抖音瀑布流字幕效果

不废话 直接上效果图 C# winform 开发抖音的瀑布流字幕。 也是typemonkey插件字幕效果 或者咱再网上常说的倒鸭子字幕效果 主要功能 1&#xff0c;软件可以自定义添加字幕内容 2&#xff0c;软件可以添加字幕显示的时间区间 3&#xff0c;可以自定义字幕颜色&#xff0c;可以随…

游戏被外挂攻破?金融数据遭篡改?AI反作弊系统实战方案(代码+详细步骤)

一、背景与需求分析 随着游戏行业与金融领域的数字化进程加速,作弊行为(如游戏外挂、金融数据篡改)日益复杂化。传统基于规则的防御手段已难以应对新型攻击,而AI技术通过动态行为分析、异常检测等能力,为安全领域提供了革命性解决方案。本文以游戏反作弊系统和金融数据安…

晶晨S905L3A(B)-安卓9.0-开启ADB和ROOT-支持IPTV6-支持外置游戏系统-支持多种无线芯片-支持救砖-完美通刷线刷固件包

晶晨S905L3A(B)-安卓9.0-开启ADB和ROOT-支持IPTV6-支持外置游戏系统-支持多种无线芯片-支持救砖-完美通刷线刷固件包 适用型号&#xff1a;M401A、CM311-1a、CM311-1sa、B863AV3.1-M2、B863AV3.2-M、UNT403A、UNT413A、M411A、E900V22C、E900V22D、IP112H等等晶晨S905L3A(B)处…

AI来了,新手如何着手学习软件开发?

AI时代新手学习软件开发的7步进化指南 &#xff08;附具体工具与避坑策略&#xff09; 一、建立“人机协作”学习观 AI是教练&#xff0c;不是替身 正确姿势&#xff1a;用AI辅助理解概念&#xff08;如让DeepSeek 、ChatGPT用生活案例解释递归&#xff09;&#xff0c;但坚持手…

《K230 从熟悉到...》矩形检测

《K230 从熟悉到...》矩形检测 《庐山派 K230 从熟悉到...》矩形检测 矩形检测技术是一种广泛应用于电子图像处理的核心技术。它通过识别和分析图像中的矩形结构&#xff0c;为各种应用提供基础支持。从传统图像处理算法到现代深度学习技术&#xff0c;矩形检测的实现途径多种多…

3. 第三放平台部署deepseek

有时候我们会发现使用deepseek服务器&#xff0c;异常卡顿&#xff0c;这是由于多方面原因造成的&#xff0c;比如说访问人数过多等。想要解决这个问题&#xff0c;我们可以选择第三方平台进行部署 第三方平台 我们可以选择的第三方平台很多&#xff0c;比如硅基流动、秘塔搜索…

【C++指针】搭建起程序与内存深度交互的桥梁(下)

&#x1f525;&#x1f525; 个人主页 点击&#x1f525;&#x1f525; 每文一诗 &#x1f4aa;&#x1f3fc; 往者不可谏&#xff0c;来者犹可追——《论语微子篇》 译文&#xff1a;过去的事情已经无法挽回&#xff0c;未来的岁月还可以迎头赶上。 目录 C内存模型 new与…

IEEE PDF Xpress校验出现 :字体无法嵌入问题以及pdf版本问题

文章目录 问题描述一、字体嵌入问题首先查看一下&#xff0c;哪些字体没有被嵌入查看window的font文件夹里的字体下载字体的网站修复字体嵌入问题 二、pdf版本不对 问题描述 在处理IEEE的camera ready的时候&#xff0c;提交到IEEE express的文件没有办法通过validate&#xf…

cookie详解

一、cookie出现原因 http是无状态的&#xff0c;浏览器无法记录当前是哪个人浏览的&#xff0c;所以出现了cookie 作用&#xff1a;会话状态管理&#xff08;用户登录状态、购物车、游戏分数&#xff09;、个性化设置&#xff08;主题、自定义设置&#xff09;、浏览器行为跟…

Mayo Clinic Platform在人工智能医疗领域的现状及启示意义研究

一、引言 1.1 研究背景与意义 在科技飞速发展的当下,人工智能(AI)已逐渐渗透至各个行业,医疗领域作为关乎人类生命健康的重要领域,也迎来了人工智能技术带来的深刻变革。人工智能医疗,作为人工智能与医疗行业深度融合的产物,正重塑着全球医疗的格局。 从全球范围来看,…

如何将 Java 应用做成 EXE 的可执行软件

目录 前言一、情景介绍二、实现步骤1. 打 Jar 包2. 编写 bat 批处理文件3. bat 转 exe 前言 最近使用 GUI 帮朋友写了一个软件&#xff0c;为了方便他处理工作上的重复性且很麻烦的事情&#xff0c;程序是使用 Java 写的&#xff0c;就不得不面对一个问题&#xff1a;我必须将…

第一篇:系统分析师首篇

目录 一、目标二、计划三、完成情况1.宏观思维导图2.过程中的团队管理和其它方面的思考 四、意外之喜(最少2点)1.计划内的明确认知和思想的提升标志2.计划外的具体事情提升内容和标志 一、目标 通过参加考试&#xff0c;训练学习能力&#xff0c;而非单纯以拿证为目的。 1.在复…

自动关机监控器软件 - 您的电脑节能助手

## 自动关机监控器 - 您的电脑节能助手 自动关机监控器是一款基于Python开发的实用工具&#xff0c;旨在帮助用户节省电力资源并延长电脑使用寿命。该程序通过监控用户的鼠标和键盘活动&#xff0c;在设定的无活动时间后自动关闭计算机&#xff0c;特别适合需要长时间离开电脑但…

线程概念与控制(中)

线程概念与控制&#xff08;上&#xff09;https://blog.csdn.net/Small_entreprene/article/details/146464905?sharetypeblogdetail&sharerId146464905&sharereferPC&sharesourceSmall_entreprene&sharefrommp_from_link我们经过上一篇的学习&#xff0c;接…

[GXYCTF2019]禁止套娃1 [GitHack] [无参数RCE]

Git基础 Git信息泄露原理解析及利用总结 - FreeBuf网络安全行业门户 CTF中的GIT泄露_ctf git泄露-CSDN博客 Git结构 dirsearch扫出来一大堆东西&#xff08;然而这些并没有什么屁用&#xff09; 但也算起码了解了git结构了吧 /.git/HEAD&#xff1a;表示当前HEAD指针的指…

从ChatGPT到AutoGPT——AI Agent的范式迁移

一、AI Agent的范式迁移 1. ChatGPT的局限性与Agent化需求 单轮对话的“工具属性” vs. 多轮复杂任务的“自主性” ChatGPT 作为强大的生成式AI,虽然能够进行连贯对话,但本质上仍然是“工具型”AI,依赖用户提供明确的指令,而无法自主规划和执行任务。 人类介入成本过高:提…

stock-pandas,一个易用的talib的替代开源库。

原创内容第841篇&#xff0c;专注智能量化投资、个人成长与财富自由。 介绍一个ta-lib的平替——我们来实现一下&#xff0c;最高价突破布林带上轨&#xff0c;和最低价突破布林带下轨的可视化效果&#xff1a; cross_up_upper stock[high].copy()# cross_up_upper 最高价突破…

Spring Cloud Gateway详细介绍简单案例

文章目录 1、Spring Cloud Gateway 详细介绍1.1. 统一入口&#xff08;Single Entry Point&#xff09;1.2. 请求路由&#xff08;Request Routing&#xff09;1.3. 负载均衡&#xff08;Load Balancing&#xff09;1.4. 流量控制&#xff08;Rate Limiting&#xff09;1.5. 身…

鸿蒙原生开发之状态管理V2

一、ArkTS状态变量的定义&#xff1a; State&#xff1a;状态&#xff0c;指驱动UI更新的数据。用户通过触发组件的事件方法&#xff0c;改变状态数据。状态数据的改变&#xff0c;引起UI的重新渲染。 在鸿蒙原生开发中&#xff0c;使用ArkTS开发UI的时候&#xff0c;我们可以…