PyTorch深度学习实战(2)——PyTorch基础
0. 前言
PyTorch
是广泛应用于机器学习领域中的强大开源框架,因其易用性和高效性备受青睐。在本节中,将介绍使用 PyTorch
构建神经网络的基础知识。首先了解 PyTorch
的核心数据类型——张量对象。然后,我们将深入研究用于张量对象的各种操作。PyTorch
提供了许多帮助构建神经网络的高级方法及组件,并提供了利用 GPU
更快地训练神经网络的张量对象。
1. 搭建 PyTorch 环境
关于 Python
的安装和配置,在此不再赘述。由于深度学习中模型的训练需要大量时间,因此通常使用 GPU
加速计算,在安装 PyTorch
之前需要根据选用的 PyTorch 版本和显卡安装 CUDA
和 cudnn
,关于 CUDA
和 cudnn
的安装和配置可以参考官方文档&#x