环境准备

本地训练

• 物理机

  如果您想在物理机上运行模型，需要进行以下环境准备：

  1. 准备昇腾硬件环境：昇腾AI服务器。
  2. 安装Python环境：建议3.7版本及以上。推荐使用Anaconda管理Python环境。
  3. ModelZoo目前支持三大深度学习框架训练与推理的模型：PyTorch，MindSpore与Tensorflow，用户可根据需要安装框架。
     • 安装PyTorch可参考：《PyTorch框架训练环境准备》
     • 安装MindSpore可参考：《昇思MindSpore安装指南》
     • 安装TensorFlow可参考：《昇腾设备TensorFlow安装指南》
• 容器

  您可以参考《Docker镜像使用教程》，下载所选模型需要的基础镜像并启动容器。启动容器后，参考模型文档安装相关依赖，即可进行模型训练与推理。

云上训练

ModelZoo部分模型适配了ModelArts云上训练平台，若由于资源限制无法进行本地开发和运行，可参考《ModelArts云上训练教程》进行云上开发和运行。