🧠 官方安装工具推荐

你可以通过 PyTorch 官方提供的安装器来获取适合你系统的命令:

👉 PyTorch 安装向导(官方推荐)


一、使用 pip 安装(适用于所有平台)

✅ CPU-only(不支持 GPU)

pip install torch torchvision torchaudio

✅ 支持 GPU(以 CUDA 11.8 为例)

确保你已安装 NVIDIA 驱动和 CUDA 11.8 工具包(或用 PyTorch 自带的 CUDA)。

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

如果你的系统支持 CUDA 12.1

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


二、使用 conda 安装(推荐方式)

如果你使用 Anaconda 环境管理器,建议用 conda 安装,更稳定。

✅ CPU-only:

conda install pytorch torchvision torchaudio cpuonly -c pytorch

✅ GPU 加速(以 CUDA 11.8 为例):

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia


三、验证安装是否成功

安装完成后,你可以在 Python 交互式终端中运行:

import torch
print(torch.__version__)
print(torch.cuda.is_available())  # 如果返回 True,表示 GPU 可用


四、安装额外组件(可选)

图像处理组件 torchvision

pip install torchvision

文本处理组件 torchtext

pip install torchtext

音频处理组件 torchaudio

pip install torchaudio


📎 出站链接(推荐)


如果你告诉我你用的是哪种操作系统和是否要用 GPU,我可以为你生成“定制安装命令”。需要吗?