PyTorch 是一个广泛使用的深度学习框架,以下是几种常见安装方式:

1. 使用 pip 安装 📦

pip install torch torchvision torchaudio

⚠️ 注意:请确保已安装 Python 环境。

pytorch_install_pip

2. 使用 conda 安装 🐍

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

🌟 详细配置可参考 Conda 安装文档

pytorch_install_conda

3. 从源码编译 🧰

git clone https://github.com/pytorch/pytorch
cd pytorch
python setup.py install

📌 需要较高技术水平,建议优先使用官方预编译版本

pytorch_build_source

验证安装 ✅

运行以下代码检查是否成功:

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

🚀 若输出版本号和 True,说明安装完成

pytorch_verify_installation

如需了解 PyTorch 基础用法,可继续阅读 PyTorch 入门教程