pytorch - PyTorch 深度学习框架
使用动态计算图和GPU加速构建和训练神经网络
标签
更新于: 2026-06-15能力
典型输入
典型输出
该技能可以做什么
- 创建张量
- 计算梯度
- 定义模型
- 加载数据集
- 训练模型
- 保存检查点
- 编译模型
- 启用分布式训练
- 导出为 TorchScript
- 导出为 ONNX
输入
- 训练数据
- 标签
- 输入维度
- 设备
- 学习率
输出
- 已训练模型
- 检查点文件
- 模型预测
- 训练损失
- TorchScript 文件
- ONNX 文件
要求
- Python
- PyTorch
- CUDA（GPU 可选）
- torchrun（分布式训练）