昇腾社区首页
中文
注册

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

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

操作步骤

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