昇腾社区首页
中文
注册

环境准备

  • Auto Tune工具运行依赖昇腾AI处理器,所以当前工具仅支持昇腾AI处理器所在设备既做开发环境又做运行环境的场景。
    图1 部署架构图
    • 请参见CANN 软件安装指南进行环境部署。
    • 请确保运行环境中执行调优用户的家目录下磁盘可用空间>=20G。
    • Auto Tune工具存储在CANN软件安装目录下的“ascend-toolkit/latest/python/site-packages/schedule_search”“ascend-toolkit/latest/python/site-packages/auto_tune”目录下。
  • 第三方软件

    环境部署完成后,需要参见表1安装Auto Tune工具依赖的第三方软件。

    表1 Auto Tune依赖软件

    需要安装的第三方软件

    用途

    如何安装

    TensorFlow 1.15

    Auto Tune工具的RL调优模式依赖TensorFlow,用于指导算子搜索。

    安装步骤请参见CANN 软件安装指南中的“安装深度学习框架 ”进行Tensorflow 1.15的安装。

    pciutils工具

    Auto Tune工具的RL调优模式使用lspci命令获取昇腾AI处理器所在硬件设备,所以需要在执行Auto Tune工具的环境中安装pciutils工具。

    例如:

    • Ubuntu操作系统:apt-get install pciutilsEulerOS操作系统:yum install pciutils

    若您是其他操作系统,请根据实际情况替换。