相关文章推荐
精明的小马驹  ·  Visual Studio ...·  1 年前    · 
温暖的书包  ·  为什么Pytorch ...·  1 年前    · 

在本教程中,我们将使用 Visual Studio Tools for AI (一个用于生成、测试和部署深度学习 & AI 解决方案的开发扩展)来训练模型。

我们将使用 Microsoft Cognitive Toolkit (CNTK) 框架和 MNIST 数据集 来训练该模型,该数据集包含 60,000 个示例的训练集和 10,000 个手写数字示例的测试集。 然后我们将使用 开放神经网络交换 (ONNX) 格式保持该模型,以与 Windows ML 一起使用。

安装 Visual Studio Tools for AI

若要开始,首先需要下载和安装 Visual Studio 。 打开 Visual Studio 之后,激活 Visual Studio Tools for AI 扩展:

  • 单击 Visual Studio 中的菜单栏,选择“扩展和更新...”
  • 单击“联机”选项卡,选择“搜索 Visual Studio Marketplace”。
  • 搜索“Visual Studio Tools for AI”。
  • 单击“下载” 按钮。
  • 安装后重启 Visual Studio。
  • Visual Studio 重启后,扩展将处于活动状态。 如果你遇到问题,请查看 查找 Visual Studio 扩展

    下载示例代码

    下载 GitHub 上的 AI 示例 存储库。 这些示例包括跨 TensorFlow、CNTK 和 Theano 等的深度学习的入门示例。

    安装 CNTK

    安装 CNTK for Python on Windows 。 请注意,如果尚未安装 Python,则需安装。

    或者,为了为深度学习模型开发准备你的计算机,请参阅 准备开发环境 ,获取用于安装 Python、CNTK、TensorFlow、NVIDIA GPU 驱动程序(可选)等的简化安装程序。

    1. 打开项目

    启动 Visual Studio,然后选择“文件”>“打开”>“项目/解决方案”。 从 AI 存储库中的示例中,选择 examples\cntk\python 文件夹,然后打开 CNTKPythonExamples.sln 文件。

    2. 训练模型

    若要将 MNIST 项目设置为启动项目,请右键单击该 python 项目,然后选择 设为启动项目

    接下来,打开 train_mnist_onnx.py 文件,按 F5 或绿色的“运行”按钮 运行 该项目。

    3. 查看模型,并将其添加到应用

    现在,经过训练的 mnist.onnx 模型文件应该放入了 samples-for-ai/examples/cntk/python/MNIST 文件夹中。

    4. 了解详细信息

    若要了解如何通过使用 Azure GPU 虚拟机 等加快培训深度学习模型的速度,请访问 Microsoft 人工智能 Microsoft 机器学习技术

    使用以下资源可获取有关 Windows ML 的帮助:

  • 若要提出或回答有关 Windows ML 的技术问题,请在 Stack Overflow 上使用 windows-machine-learning 标记。
  • 若要报告 bug,请在 GitHub 上提交问题。
  •