PyTorch ML - PyTorch深度学习环境
配备CUDA的PyTorch机器学习环境，支持模型训练和GPU加速
标签
更新于: 2026-03-21能力
典型输入
典型输出
该技能可以做什么
- 定义神经网络
- 加载预训练模型
- 训练模型
- 执行推理
- 加载数据
- 预处理数据
- 可视化训练
- GPU加速
- 保存检查点
- 运行分布式训练
- 集成NumPy
- 集成Pandas
- 集成scikit-learn
输入
- 训练数据集
- 模型架构
- 超参数
- 预训练模型
输出
- 模型检查点
- 训练指标
- 推理结果
- 训练可视化
要求
- CUDA支持
- GPU