学习torch-geometric时,在pycharm中直接run运行代码发现报错:
OSError: libcusparse.so.11.0: cannot open shared object file: No such file or directory
找不到cuda相关文件。
首先尝试定位问题,
第一,在terminal中echo相关环境变量如$PATH$LD_LIBRARY_PATH等,结果显示均配置正常。
第二,检查所运行torch框架及各种扩展包与cuda版本匹配。
无法解决后进行搜索,发现原因是由于pycharm找不到cuda,而与系统无关。

解决办法(不知是否彻底)是在终端:sudo ldconfig /usr/local/cuda/lib64

references:https://blog.csdn.net/baidu_40840693/article/details/108769679

Logo

欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐