🧠 官方安装工具推荐
你可以通过 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
📎 出站链接(推荐)
- 官方安装文档:https://pytorch.org/get-started/locally/
- CUDA 工具包下载:https://developer.nvidia.com/cuda-downloads
- Conda(Anaconda)官网:https://www.anaconda.com/
如果你告诉我你用的是哪种操作系统和是否要用 GPU,我可以为你生成“定制安装命令”。需要吗?
发表回复