torch-scatter import时候的问题version_cuda.so

torch-scatter import时候的问题

torch_scatter/_version_cuda.so: undefined symbol: ZN3c106detail12infer_schema20make_function_schemaENS_8ArrayRefINS1_11ArgumentDefEEES4

遇到了以上的报错,该报错是scatter,从报错的文件名不难猜测出事关于版本的问题,这个问题的具体原因也便是因为你主机上的与torch相关的文件版本不适配,所以我们需要将有关torch的版本调为统一。
pytorch网站,根据该网站,我们将torch-scatter torch-sparse torch-cluster torch-spline-conv torch-geometric几个库更新为与我们电脑上torch版本一致。
查询torch,cuda版本代码:

import torch
print(torch.__version__)

print(torch.version.cuda)
print(torch.backends.cudnn.version())

pip install torch-scatter torch-sparse torch-cluster torch-spline-conv torch-geometric -f https://data.pyg.org/whl/torch-{version}+cu{version}.html

torch-scatter import时候的问题version_cuda.so_第1张图片
问题解决。

你可能感兴趣的:(pip,python)