昇腾社区首页
中文
注册

升级

用户如需将当前版本的Rec SDK Torch升级至最新版本,可将最新的Rec SDK Torch软件包上传至安装环境后,在软件包所在目录下使用命令进行升级,具体命令如下。

  • 升级Rec SDK Torch新版本时,需要先手动卸载旧版本。
    # 卸载hybrid_torchrec
    pip3 uninstall hybrid_torchrec -y
    # 卸载torchrec
    pip3 uninstall torchrec -y
    # 卸载算子
    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
  • 参考安装Rec SDK Torch重新安装Rec SDK Torch