AI Rookies

PyTorch — 深度学习框架 PyTorch

事实

一个用于构建和训练神经网络的深度学习框架。

人话

PyTorch 像实验室白板,公式刚写一半就能上手改;哪层不对,研究员擦了就重来。

它适合快速搭模型、做实验和训练网络,很多论文原型都先在这儿落地。

相关概念

Framework
PyTorch 是深度学习框架的一种,主打灵活开发与训练。

Automatic Differentiation
它内置自动求导,能自动计算训练所需梯度。

GPU
PyTorch 常把计算放到 GPU 上,加快模型训练与实验。

Cuda
它经常配合 CUDA 使用,把模型计算交给显卡执行。