使用 PyTorch 进行深度学习:60 分钟闪电战¶
创建时间: 2017 年 3 月 24 日 |最后更新时间:2023 年 5 月 31 日 |上次验证: Nov 05, 2024
作者: Soumith Chintala
什么是 PyTorch?¶
PyTorch 是一个基于 Python 的科学计算包,主要有两大用途:
NumPy 的替代品,以使用 GPU 和其他加速器的强大功能。
一个用于实现神经网络的自动微分库。
本教程的目标:¶
大致了解 PyTorch 的 Tensor 库和神经网络。
训练小型神经网络以对图像进行分类
要运行下面的教程,请确保您拥有 Torch、torchvision、 和 matplotlib 软件包。
在本教程中,您将学习 PyTorch 张量的基础知识。
了解 autograd。
本教程演示了如何在 PyTorch 中训练神经网络。
了解如何在 PyTorch 中使用 CIFAR10数据集。