问题分析

pip安装完pytorch=1.13.1后,build代码时出错:

 undefined symbol: cublasLtHSHMatmulAlgoInit, version libcublasLt.so.11

原因因为pytorch1.13安装时会给我们自动安装

nvidia_cublas_cu11,nvidia_cuda_nvrtc_cu11,nvidia_cuda_runtime_cu11 and nvidia_cudnn_cu11. 

但由于一般用户已经安装CUDAtoolkit,所以会产生这个错误。

解决方法

卸载默认安装的nvidia包

pip uninstall nvidia_cublas_cu11


————————————————
参考:https://blog.csdn.net/be_humble/article/details/129038771

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐