windows安装pytorch指令

老刘

Windows 安装 PyTorch 指令详解:从零到跑通环境

如果你正在找 Windows 安装 PyTorch 指令,大概率是想尽快把深度学习环境搭起来,开始跑代码、训练模型或者测试 GPU 是否可用。
这篇文章直接按“问题分析 → 解决方案 → 注意事项”的方式讲清楚,适合新手一步一步照着做,也能给进阶用户提供更高效的命令方案。


一、先搞清楚:Windows 上装 PyTorch,核心看这 3 件事

很多人一上来就复制安装命令,结果不是装不上,就是版本冲突。其实安装 PyTorch 之前,先确认这 3 个关键点:

1. 你的 Python 版本是否合适

PyTorch 对 Python 版本有要求,常见稳定组合是:

  • Python 3.9
  • Python 3.10
  • Python 3.11

一般不建议一上来就用太新的版本,否则容易遇到依赖不兼容。

2. 你是装 CPU 版还是 GPU 版

这个决定了你的安装指令:

  • CPU 版:不需要 NVIDIA 显卡,装起来最简单,适合学习、测试、轻量开发。
  • GPU 版:需要 NVIDIA 显卡 + 对应 CUDA 支持,适合训练模型、加速推理。

3. 你是否已经装好环境管理工具

推荐两种方式:

  • Conda:最稳,适合新手
  • venv + pIP:纯 Python 环境,适合轻量用户

如果你是第一次装,建议优先用 Conda


二、最推荐的安装方案:Conda 安装 PyTorch

方案一:安装 CPU 版 PyTorch

如果你只是想先把环境跑通,建议直接装 CPU 版。

1)先创建环境

打开 Anaconda Prompt,输入:

conda create -n pytorch python=3.10 -y
conda activate pytorch

2)安装 PyTorch CPU 版

执行:

conda install pytorch torchvision torchaudio cpuonly -c pytorch -y

3)验证是否安装成功

输入:

python

进入 Python 后,输入:

import torch
print(torch.__version__)
print(torch.cuda.is_available())

如果返回版本号,且 torch.cuda.is_available() 输出 False,这很正常,因为你装的是 CPU 版。


方案二:安装 GPU 版 PyTorch

如果你电脑有 NVIDIA 显卡,并且想用 GPU 加速,推荐装 GPU 版。

1)先创建环境

conda create -n pytorch python=3.10 -y
conda activate pytorch

2)安装支持 CUDA 的 PyTorch

常用安装方式如下:

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia -y

这个命令的意思是:

  • 安装 PyTorch
  • 安装图像处理库 torchvision
  • 安装音频库 torchaudio
  • 指定 CUDA 12.1 运行环境

3)验证 GPU 是否可用

安装完成后,进入 Python 执行:

import torch
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.cuda.get_device_name(0))

如果:

  • torch.cuda.is_available() 返回 True
  • torch.cuda.get_device_name(0) 能显示显卡名称

说明 GPU 环境已经正常。


三、如果你用的是 pip,安装指令也很简单

有些人不想装 Conda,也可以直接用 pip

方案一:pip 安装 CPU 版

先升级 pip:

python -m pip install --upgrade pip

然后安装:

pip install torch torchvision torchaudio

这个方式适合快速尝试,但在 Windows 上,有时 Conda 会更稳定。


方案二:pip 安装 GPU 版

如果你想用 pip 安装 GPU 版,建议去 PyTorch 官方对应页面选择命令。
常见形式类似下面这样:

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

如果你使用的是其他 CUDA 版本,链接会不同,比如 cu118cu124 等。


四、Windows 新手最稳的完整安装流程

如果你完全是小白,建议直接照着下面做,成功率最高。

第一步:安装 Anaconda

安装好 Anaconda 后,会自带 conda 命令。

第二步:打开 Anaconda Prompt

不要用普通 CMD 也可以,但 Anaconda Prompt 更省事。

第三步:创建独立环境

conda create -n pytorch python=3.10 -y
conda activate pytorch

第四步:安装 PyTorch

CPU 版:

conda install pytorch torchvision torchaudio cpuonly -c pytorch -y

GPU 版:

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia -y

第五步:测试

python

然后输入:

import torch
print(torch.__version__)
print(torch.cuda.is_available())

五、常见问题分析与解决办法

问题 1:torch 导入报错

比如:

ModuleNotFoundError: No module named 'torch'

原因

通常是:

  • 没装成功
  • 装到了另一个环境
  • 当前 Python 和安装时的 Python 不是同一个

解决方案

先检查环境:

conda activate pytorch
python -m pip show torch

如果查不到,再重新安装一次对应版本。


问题 2:GPU 版装了但 torch.cuda.is_available() 还是 False

原因

可能是以下几种:

  • 显卡不是 NVIDIA
  • 驱动没装好
  • CUDA 环境不匹配
  • 装成了 CPU 版

解决方案

先确认显卡型号:

nvidia-smi

如果这个命令都没法正常显示显卡信息,说明 NVIDIA 驱动可能没装好。
建议先升级显卡驱动,再重装 PyTorch。


问题 3:conda 安装很慢

原因

国内网络访问官方源速度一般不快。

解决方案

可以换镜像源,但 PyTorch 和 CUDA 相关包有时还是建议优先用官方源,避免版本不一致。
如果你想提高速度,可以先配置 Conda 镜像,再安装常规包;PyTorch 安装时尽量保持来源一致。


问题 4:pip 装完版本不对

原因

pip 可能从默认源下载了不匹配的包。

解决方案

强制指定官方 PyTorch 下载地址,比如:

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

六、不同需求下,怎么选安装方式

1)只学习、写 Demo、跑 CPU 推理

推荐:

conda install pytorch torchvision torchaudio cpuonly -c pytorch -y

优点:

  • 简单
  • 稳定
  • 兼容性好

2)有 NVIDIA 显卡,想训练模型

推荐:

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia -y

优点:

  • 安装相对省心
  • GPU 支持好
  • 适合长期开发

3)临时测试、轻量使用

可以用:

pip install torch torchvision torchaudio

优点:

  • 简单

缺点:

  • 环境管理不如 Conda 稳

七、安装前的注意事项

1. 尽量不要把多个 Python 环境混着用

比如:

  • 系统自带 Python
  • Anaconda Python
  • Microsoft Store Python

混用最容易出问题。
建议明确只用一个环境。

2. 不建议直接装在系统 Python 里

尤其是新手,最好用虚拟环境隔离,避免污染系统。

3. GPU 版要先确认驱动正常

如果驱动不正常,PyTorch 装得再对也用不了 GPU。

4. 不要盲目追最新版本

很多时候稳定比“最新”更重要。
如果你是做课程、项目、论文复现,优先用常见稳定版本。


八、最常用的几条 PyTorch 安装指令汇总

1)Conda CPU 版

conda create -n pytorch python=3.10 -y
conda activate pytorch
conda install pytorch torchvision torchaudio cpuonly -c pytorch -y

2)Conda GPU 版

conda create -n pytorch python=3.10 -y
conda activate pytorch
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia -y

3)pip CPU 版

python -m pip install --upgrade pip
pip install torch torchvision torchaudio

4)pip GPU 版

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

九、安装成功后的快速验证代码

安装完后,建议直接运行下面这段测试代码:

import torch

print("PyTorch版本:", torch.__version__)
print("CUDA是否可用:", torch.cuda.is_available())

if torch.cuda.is_available():
    print("显卡名称:", torch.cuda.get_device_name(0))

如果能正常输出,就说明 PyTorch 环境已经可以用了。


十、结论:Windows 安装 PyTorch 最稳的做法

如果你想少踩坑,直接记住这句话就够了:

  • 新手优先用 Conda
  • 没有 NVIDIA 显卡就装 CPU 版
  • 有 NVIDIA 显卡就装 GPU 版
  • Python 推荐 3.10
  • 安装后一定要测试 import torchtorch.cuda.is_available()

最稳的 GPU 安装命令就是:

conda create -n pytorch python=3.10 -y
conda activate pytorch
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia -y

如果你是纯学习入门,CPU 版命令则更省心:

conda create -n pytorch python=3.10 -y
conda activate pytorch
conda install pytorch torchvision torchaudio cpuonly -c pytorch -y

这两套命令基本能覆盖 Windows 下大多数安装需求。

文章版权声明:文章内容均来源于各大短视频平台搜集以及修改和删减新增,如有侵权或者违规,请联系站长进行删除,如需转载或复制请以超链接形式并注明出处。

发表评论

快捷回复: 表情:
AddoilApplauseBadlaughBombCoffeeFabulousFacepalmFecesFrownHeyhaInsidiousKeepFightingNoProbPigHeadShockedSinistersmileSlapSocialSweatTolaughWatermelonWittyWowYeahYellowdog
验证码
评论列表 (暂无评论,3人围观)

还没有评论,来说两句吧...

目录[+]