PyTorch 是目前深度学习最主流的框架之一,灵活易用,广泛用于学术与工业界。
本篇将带你从环境安装开始,逐步掌握 PyTorch 最核心的两个部分:
张量(Tensor)
:神经网络的基本数据结构自动求导(Autograd)
:支持反向传播的魔法核心
🧱 一、环境安装方式(含 Conda / Pip / CUDA)
✅ 推荐安装方式:conda(稳定、自动管理依赖)
# 安装 CPU 版本(适合没有显卡或初学者)
conda install pytorch torchvision torchaudio cpuonly -c pytorch
# 安装 GPU 版本(自动识别 CUDA)
conda