昇腾社区首页
中文
注册
开发者
下载

卸载

用户如需移除Rec SDK Torch软件包,可参考以下命令进行卸载。

# 卸载hybrid_torchrec
pip3 uninstall hybrid_torchrec -y
# 卸载 torchrec_embcache
pip3 uninstall torchrec_embcache -y
# 卸载torchrec
pip3 uninstall torchrec -y

用户如需移除Rec SDK Torch自定义算子相关包,可参考以下命令进行卸载。其中,

  • 自定义算子在CANN中的默认安装路径为/usr/local/Ascend/ascend-toolkit/latest/opp/vendors/
  • 卸载自定义算子时,删除vendors路径下自定义算子名称对应的文件夹即可。可以通过ai_core_op查看Rec SDK的自定义算子目录。
# 卸载算子指令示例
rm -rf /usr/local/Ascend/ascend-toolkit/latest/opp/vendors/asynchronous_complete_cumsum
rm -rf /usr/local/Ascend/ascend-toolkit/latest/opp/vendors/backward_codegen_adagrad_unweighted_exact
rm -rf /usr/local/Ascend/ascend-toolkit/latest/opp/vendors/permute2d_sparse_data
rm -rf /usr/local/Ascend/ascend-toolkit/latest/opp/vendors/split_embedding_codegen_forward_unweighted
# 卸载libfbgemm_npu_api.so
PACKAGE_PATH=$(python3 -c "import sysconfig; print(sysconfig.get_path('purelib'))")
if [ -d "$PACKAGE_PATH" ]; then
  cd ${PACKAGE_PATH}
  rm -rf libfbgemm_npu_api.so
else
  echo "no site-package"
fi

部分自定义算子安装到CANN的vendors文件夹下时,原自定义算子名称与安装路径不同,卸载时需删除安装后的路径。存在差异的自定义算子如下表。

自定义算子名称

安装到vendors文件夹下的名称

cust_op_by_addr

customize

pcie_through

customize

fused_lazy_adam

mxrec_fused_lazy_adam

fused_sgd

mxrec_sgd

用户自定义算子在未配置安装文件夹名称时,默认安装在CANN的vendors/customize文件夹下。当删除的customize文件夹中存在用户自定义算子时,则对应算子需重新安装。