卸载
用户如需移除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文件夹中存在用户自定义算子时,则对应算子需重新安装。
父主题: 安装部署