安装框架插件包

在线推理或训练场景下若使用深度学习框架TensorFlow,需要安装框架插件包。包括如下两种方式:

方式一:不定制,请直接使用run包安装

  1. 获取Ascend-cann-tfplugin_xxx.run。
  2. 请参照安装开发套件包的步骤安装框架插件包。软件包安装详细日志路径如下。
    • root用户:“/var/log/ascend_seclog/ascend_tfplugin_install.log”
    • 非root用户:“${HOME}/var/log/ascend_seclog/ascend_tfplugin_install.log”

      其中${HOME}为当前用户目录。

  3. 配置环境变量。

    CANN软件提供进程级环境变量设置脚本,供用户在进程中引用,以自动完成环境变量设置。用户进程结束后自动失效。示例如下(以root用户默认安装路径为例):

    . /usr/local/Ascend/tfplugin/set_env.sh
    用户也可以通过修改~/.bashrc文件方式设置永久环境变量,操作如下:
    1. 以运行用户在任意目录下执行vi ~/.bashrc命令,打开.bashrc文件,在文件最后一行后面添加上述内容。
    2. 执行:wq!命令保存文件并退出。
    3. 执行source ~/.bashrc命令使其立即生效。

方式二:需要定制,请通过源码包方式编译安装

  1. 下载框架插件源码包。示例命令如下:
    git clone -b tfa_v0.0.20_7.0.0.alpha003 https://gitee.com/ascend/tensorflow.git
  2. 参见下载包中的“README.md”进行修改、编译和安装。