游乐游手机版
首页/AI教程/文章详情

人工智能PyTorch安装教程新手入门

时间:2026-06-02 12:10
准备开启 PyTorch 的学习旅程了吗?对于刚入门的新手来说,安装环境往往是第一道障碍。别担心,这份详细指南将手把手带你完成 PyTorch 安装,并避开常见的“坑”。实际上,整个过程远比你想象得简单。 PyTorch 安装指南(新手必看) 安装前的准备工作 在正式安装 PyTorch 之前,有两

准备开启 PyTorch 的学习旅程了吗?对于刚入门的新手来说,安装环境往往是第一道障碍。别担心,这份详细指南将手把手带你完成 PyTorch 安装,并避开常见的“坑”。实际上,整个过程远比你想象得简单。

PyTorch 安装指南(新手必看)

安装前的准备工作

在正式安装 PyTorch 之前,有两项基础准备工作必须做好,这能确保后续操作顺利无阻。

  1. 安装 Python 环境

    PyTorch 要求 Python 版本在 3.6 及以上(Python 2 已不再支持)。请先检查你的电脑是否已安装符合条件的 Python。

    检查方法:打开命令行工具(Windows 用户使用 CMD 或 PowerShell,Mac/Linux 用户使用终端),输入 python --versionpython3 --version。如果显示的版本号高于 3.6,说明已满足要求。

    若尚未安装,请前往 Python 官网下载安装包。安装时,务必勾选“Add Python to PATH”选项,这样可以避免后续手动配置环境变量的麻烦。

  2. 选择包管理工具

    强烈推荐使用 Anaconda。它不仅是 Python 发行版,更是一个强大的包与环境管理器,能帮你轻松管理项目依赖,避免版本冲突。对数据科学和机器学习新手而言,这几乎是必备工具。

    从 Anaconda 官网下载适合你操作系统的安装包,安装过程中同样记得勾选“Add Anaconda to my PATH environment variable”(或类似选项)。


安装 PyTorch 框架

准备工作就绪后,现在可以正式安装 PyTorch 了。这里提供两种主流方式,你可以根据偏好选择。

方法一:使用 Anaconda(推荐)

  1. 打开 Anaconda Prompt(Windows)或终端(Mac/Linux)。这是执行 Conda 命令的正确环境。
  2. 创建虚拟环境(可选但强烈推荐)

    这是一个极好的习惯。为 PyTorch 项目创建独立的虚拟环境,可以确保它的依赖库不影响电脑上其他 Python 项目。执行以下命令:

    conda create -n pytorch_env python=3.8  # 创建名为 pytorch_env、Python 3.8 的环境
    conda activate pytorch_env               # 激活该环境

    后续所有操作都将在 pytorch_env 环境内进行。

  3. 获取安装命令

    这是最关键的一步。访问 PyTorch 官方安装页面,你会看到一个配置生成器。

    • Package:选择 Conda(如果你用 Anaconda)或 Pip。
    • Compute Platform:如果你的电脑没有 NVIDIA GPU,选择“CPU”;如果有,则根据显卡驱动选择对应的 CUDA 版本(需要提前安装 NVIDIA 驱动和 CUDA Toolkit)。

    选择完成后,网站会生成对应的安装命令。例如,常见的 CPU 版本安装命令如下:

    conda install pytorch torchvision torchaudio cpuonly -c pytorch

    将生成的命令复制,粘贴到 Anaconda Prompt 或终端中执行,等待安装完成即可。

方法二:使用 Pip

如果你更偏好轻量的 Pip,同样可行。在 PyTorch 官网配置生成器中选择“Pip”和你的平台,它会给出对应的 pip install 命令。例如:

pip install torch torchvision torchaudio

在命令行中执行该命令即可。注意,使用 Pip 时,也建议在虚拟环境中操作,可以使用 Python 自带的 venv 模块创建虚拟环境。


不同平台的注意事项

  • Windows 用户:安装过程中若遇到与 C++ 库相关的错误,可能需要安装“Microsoft Visual C++ Redistributable”,可从微软官网免费下载。
  • Mac/Linux 用户:过程通常更简单,一般无需额外配置。Mac 用户请注意,PyTorch 对 Apple Silicon 芯片(M1/M2 等)的原生支持在较新版本中才完善,请确认官网命令是否包含 -c pytorch 以外的 -c conda-forge 等特定渠道。

验证安装是否成功

安装完成后,务必验证一下。打开 Python 环境(如果使用了虚拟环境,请确保已激活),输入以下几行代码:

import torch
print(torch.__version__)          # 查看已安装的 PyTorch 版本号
print(torch.cuda.is_a vailable())  # 输出 True 表示 GPU 可用,False 表示仅 CPU

如果一切正常,你会看到类似下面的输出:

2.1.0        # 你的版本号可能不同
False        # 如果安装的是 CPU 版本,这里就是 False

看到版本号且没有报错,恭喜你,PyTorch 已成功安装!


常见问题与解决方案

新手安装时可能会遇到一些小波折,这里列举几个典型问题及其解决方法。

  1. 提示“conda/pip 不是内部或外部命令”

    这通常意味着 Anaconda 或 Python 没有正确添加到系统环境变量 PATH 中。请检查安装步骤中是否勾选了“Add to PATH”选项,并尝试重启命令行工具或计算机。如果问题依旧,可能需要手动将安装路径添加到 PATH 中。

  2. 下载速度慢或安装失败

    由于网络原因,从官方源下载可能很慢甚至超时。解决办法是更换为国内镜像源,速度会大幅提升。

    对于 Conda,可以配置清华镜像源:

    conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
    conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
    conda config --set show_channel_urls yes

    对于 Pip,可以在安装命令后指定镜像源:

    pip install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple
  3. 导入 torch 时报错

    如果执行 import torch 时出现错误,首先检查 Python 版本是否与 PyTorch 版本兼容(官网配置器会自动匹配好)。另一个常见原因是安装不完整或存在冲突,可以尝试先卸载再重新安装:

    conda uninstall pytorch torchvision torchaudio
    # 然后重新执行官网提供的安装命令

学习资源推荐

安装只是第一步,接下来就是深入学习了。这里有几个公认的优秀起点:

  • 官方教程:PyTorch 官方文档的 Tutorials 部分是最权威的入门材料,系统且严谨。
  • 经典书籍:《动手学深度学习》(Dive into Deep Learning)提供了 PyTorch 实现版本,理论与实践结合出色,非常适合新手。
  • 实践项目:不要只看不练。从最简单的线性回归、MNIST 手写数字识别项目开始,亲手编写代码、运行模型,是巩固知识最快的方式。

现在,环境已经搭建完成,学习路径也已清晰。接下来,打开你的编辑器,开始探索深度学习的奇妙世界吧。

来源:https://blog.csdn.net/u012392578/article/details/146947752
上一篇ChatGPT数据可视化:上传文件一键出图已成现实 下一篇AI图片生成新手零失败无版权风险教程
本站内容用于信息整理与展示,如有侵权或内容问题请及时联系处理。

相关推荐

补充同频道和同主题内容,方便继续浏览更多相关内容。

同类最新

继续查看同栏目最近更新的文章。

更多
Claude Design提示词3步复刻同款Agent教程
AI教程 · 2026-06-02

Claude Design提示词3步复刻同款Agent教程

Anthropic的ClaudeDesign设计智能体提示词被复刻至CodeBuddy子智能体,实现海报、PPT、交互原型、动效、UI设计系统及多格式导出等多种设计功能。用户仅需三步配置即可使用,充分彰显了提示词工作流的强大效果,显著提升设计效率。

CloudQ如何将故障根因分析时间从45分钟缩短至6分钟
AI教程 · 2026-06-02

CloudQ如何将故障根因分析时间从45分钟缩短至6分钟

摘要:传统故障根因分析(RCA)通常耗时45至105分钟,而CloudQ借助WorkBuddy与CloudQ双引擎协同、架构感知能力、多产品日志聚合以及异常模式识别技术,将RCA过程大幅压缩至约6分钟,一线值班人员通过即时通讯工具即可完成首轮根因分析。一、传统 RCA 的“时间黑洞”故障发生时,每一

Go语言开发AI Agent的关键原因与核心优势全面解析
AI教程 · 2026-06-02

Go语言开发AI Agent的关键原因与核心优势全面解析

Go语言借助轻量级goroutine实现高并发与低内存占用,采用通道通信替代共享内存,天然适配长时间运行的AIAgent进程,可高效支撑数千并发任务,显著降低基础设施成本。

告别低效科研,轻松完成专业算力分析
AI教程 · 2026-06-02

告别低效科研,轻松完成专业算力分析

科研人员大量时间耗费在数据清洗、数值运算等基础琐事,普通AI仅能文字辅助,无法处理核心科研。AiPy依托Python生态,无需编程即可完成数据分析、模型拟合、文献梳理、仿真模拟,支持本地离线运行,保障科研机密。

GPT-5.5写A/B测试文案效率提升镜像实验复盘
AI教程 · 2026-06-02

GPT-5.5写A/B测试文案效率提升镜像实验复盘

通过镜像实验对比人工、普通大模型与GPT-5 5生成A B测试文案,结果显示GPT-5 5可将初稿生成时间从2小时压缩至20分钟,并显著提升策略区分度、变量控制能力及复盘效率,推动文案实验向工程化转型。