使用 PyTorch 进行深度学习:60 分钟快速入门
作者: Soumith Chintala
什么是 PyTorch?
PyTorch 是一个基于 Python 的科学计算库,主要服务于两大用途:
-
一个替代 NumPy 的工具,能够利用 GPU 和其他加速器的强大性能。
-
一个自动微分库,对于实现神经网络非常有用。
本教程的目标:
-
从高层次理解 PyTorch 的 Tensor 库和神经网络。
-
训练一个小的神经网络来进行图像分类
要运行以下教程,请确保您已安装 torch、torchvision 和 matplotlib 包。