在线推理或训练场景下若使用深度学习框架TensorFlow,需要安装框架插件包。包括如下两种方式:
方式一:不定制,请直接使用run包安装
- 获取Ascend-cann-tfplugin_xxx.run。
- 请参照安装开发套件包的步骤安装框架插件包。软件包安装详细日志路径如下。
- root用户:“/var/log/ascend_seclog/ascend_tfplugin_install.log”
- 非root用户:“${HOME}/var/log/ascend_seclog/ascend_tfplugin_install.log”
其中${HOME}为当前用户目录。
- 配置环境变量。
CANN软件提供进程级环境变量设置脚本,供用户在进程中引用,以自动完成环境变量设置。用户进程结束后自动失效。示例如下(以root用户默认安装路径为例):
. /usr/local/Ascend/tfplugin/set_env.sh
用户也可以通过修改~/.bashrc文件方式设置永久环境变量,操作如下:
- 以运行用户在任意目录下执行vi ~/.bashrc命令,打开.bashrc文件,在文件最后一行后面添加上述内容。
- 执行:wq!命令保存文件并退出。
- 执行source ~/.bashrc命令使其立即生效。
方式二:需要定制,请通过源码包方式编译安装
- 下载框架插件源码包。示例命令如下:
git clone -b tfa_v0.0.20_7.0.0.alpha003 https://gitee.com/ascend/tensorflow.git
- 参见下载包中的“README.md”进行修改、编译和安装。