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

1. 从源码安装 🧰

git clone https://github.com/pytorch/pytorch.git
cd pytorch

⚠️ 需要提前安装 CMake 和其他依赖项

2. 使用 Conda 环境 🐍

conda create -n pytorch_env python=3.9
conda activate pytorch_env
conda install pytorch torchvision torchaudio cudatoolkit -c pytorch

3. 使用 Pip 安装 📦

pip install torch torchvision torchaudio

📌 请确保已配置好 虚拟环境 管理

4. 验证安装 ✅

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

📌 扩展阅读PyTorch 快速入门

PyTorch_Installation