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

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

操作步骤

  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. 解压压缩包,进入“mxrec-branch_v6.0.0/cust_op/cust_op_by_addr”路径下,参考以下命令编译并安装片上内存显存侧动态扩容算子包。

    1
    bash run.sh