【2025最新】PyTorch安装全攻略:CPU/GPU版一步到位(Windows/macOS/Linux)

摘要:本文手把手教你安装PyTorch深度学习框架,涵盖Anaconda/pip两种安装方式,详细讲解CPU与GPU版本选择、CUDA配置及环境验证,解决99%的安装报错问题!


一、PyTorch简介

PyTorch是Meta(原Facebook)开源的深度学习框架,以其动态计算图和易用性广受研究人员喜爱。支持GPU加速训练,是AI开发者的首选工具。


二、安装前准备

  1. 硬件检查
    • 确认显卡型号(NVIDIA显卡需支持CUDA)
    • Windows用户安装NVIDIA驱动
  2. 查看CUDA版本(GPU用户):
    nvidia-smi  # 右上角显示最高支持的CUDA版本
    
  3. Python环境:需3.7+版本(建议通过Anaconda管理)

三、通过Anaconda安装(推荐)

▶ 步骤1:创建虚拟环境

conda create -n pytorch_env python=3.9
conda activate pytorch_env

▶ 步骤2:访问PyTorch官网获取安装命令

  1. 打开官网:https://pytorch.org/
  2. 选择配置:
    • PyTorch版本:稳定版(Stable)
    • 操作系统:根据实际情况选择
    • 包管理器:Conda/Pip
    • 语言:Python
    • CUDA:选

猜你喜欢

转载自blog.csdn.net/weixin_69882801/article/details/145541154
今日推荐