Pytorch介绍与安装

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