1、安装
官网:https://pytorch.org/
# 验证是否安装成功以及相关版本
import torch
print(torch.__version__)
# 检查cuda是否可以使用
print(torch.cuda.is_available())
2、介绍
torch:pytorch的核心库,提供张量操作、自动微分系统、优化器等基础功能,提供了大量的数学运算与张量操作,是构建深度学习模型的基础。
torchvison:pytorch的一个拓展库,专门用于处理图像数据,提供了许多预训练模型(ResNet、VGG等),可直接用于图像分类、目标检测等任务。
torchaudio:pytorch的音频处理库,用于处理音频数据。