报错:OSError: libcublas.so.10: cannot open shared object file: No such file or directory

具体地

服务器环境中已经安装了cuda102版的torch,也安装了dgl-cu102包,但是在导入dgl时

import dgl

这里报错,错误为:

OSError: libcublas.so.10: cannot open shared object file: No such file or directory

解决方案

环境中未安装cudatoolkit,通过以下命令安装10.2版本的

conda install cudatoolkit=10.2 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/

再次import,不再报错,问题解决

你可能感兴趣的:(BUG及解决方案,pytorch,python)