背景提要:刚刚安装了pytorch,torch-geometric、torch_scatter和torch_sparse,在调用语句from torch_geometric.data import Data出现了问题
错误代码:OSError: libcusparse.so.10: cannot open shared object file: No such file or directory
可能原因:anaconda没有与cudatoolkit的环境路径绑定在一起
解决方法:conda install cudatoolkit=10.2,后面的版本需要根据自己的cuda版本进行更改,再次测试就成功啦。
pytorch 测试成功完结撒花,耶。
网友评论