文档
注册

环境准备

  • 使用ModelSlim工具前,需参考CANN 软件安装指南搭建开发环境。
  • 安装CANN软件后,需要以CANN运行用户登录环境,执行如下示例命令配置环境变量。
    source {CANN包安装路径}/ascend-toolkit/set_env.sh
  • 使用非root用户运行调优任务时,需要管理员用户将运行用户加入驱动运行用户组(例如:HwHiAiUser)中,保证普通用户对run包的lib库有读权限。
  • ModelSlim工具依赖Python,以Python3.7.5为例,请以运行用户执行如下命令设置Python3.7.5的相关环境变量。
    #用于设置python3.7.5库文件路径
    export LD_LIBRARY_PATH=/usr/local/python3.7.5/lib:$LD_LIBRARY_PATH
    #如果用户环境存在多个python3版本,则指定使用python3.7.5版本
    export PATH=/usr/local/python3.7.5/bin:$PATH

    Python3.7.5安装路径请根据实际情况进行替换,您也可以将以上命令写入~/.bashrc文件中,然后执行source ~/.bashrc命令使其立即生效。

  • 用户根据需要自行安装以下AI框架包(请注意CANN与MindSpore、PyTorch的版本配套关系):
    • MindSpore:

      请参考MindSpore官网安装MindSpore框架。

    • PyTorch:
      • 请参考安装PyTorch,安装PyTorch框架、Torchvision依赖和Apex混合精度模块。
      • 若PyTorch下需要统计模型的参数量信息,则执行如下命令安装依赖thop。
        如下命令如果使用非root用户安装,需要在安装命令后加上--user,例如:pip3 install thop --user
        pip3 install --user thop
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词