昇腾社区首页
中文
注册

(可选)片上内存显存侧动态扩容算子包安装

如需使用片上内存显存侧动态扩容功能,请参考本步骤编译安装片上内存显存侧动态扩容算子包。

操作步骤

  1. 执行如下命令,在容器中设置CANN相关环境变量。
    source /usr/local/Ascend/ascend-toolkit/set_env.sh
    source /usr/local/Ascend/tfplugin/set_env.sh
  2. 进入已解压的Rec SDK软件包“mindxsdk-mxrec/cust_op/cust_op_by_addr”目录中,参考以下命令编译并安装片上内存显存侧动态扩容算子包。
    bash run.sh