Pytorch介绍与安装

1、安装

官网：https://pytorch.org/

# 验证是否安装成功以及相关版本
import torch

print(torch.__version__)
# 检查cuda是否可以使用
print(torch.cuda.is_available())

2、介绍

torch：pytorch的核心库，提供张量操作、自动微分系统、优化器等基础功能，提供了大量的数学运算与张量操作，是构建深度学习模型的基础。

torchvison：pytorch的一个拓展库，专门用于处理图像数据，提供了许多预训练模型（ResNet、VGG等），可直接用于图像分类、目标检测等任务。

torchaudio：pytorch的音频处理库，用于处理音频数据。