python安装torch-geometric包出现的错误:OSError: torch_sparse/_version_cuda.so: undefined symbol:

python安装torch-geometric包出现的错误:OSError: /home/zh1995/anaconda3/lib/python3.6/site-packages/torch_sparse/_version_cuda.so: undefined symbol: _ZN3c106detail23torchInternalAssertFailEPKcS2_jS2_S2_

    • 原因:
    • 解决方案:
    • 具体解决步骤:

原因:

这个错误通常是安装了多个PyTorch或多个CUDA版本的原因。通过anaconda-navigator安装pytorch 时它将自动安装cpu和gpu两个版本,因此产生错误。

解决方案:

不要通过anaconda来安装Torch,而是通过pip从以下网址下载特定版本的pytorch: https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html。

具体解决步骤:

  1. 卸载原有的包:pip uninstall 相应的依赖包(如torch、torch-scatter、torch-sparse)

  2. 通过pip安装指定cuda版本的torch:pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
    此处我安装的torch是1.8,cuda是11.1,也可以安装别的版本,但是一定要通过pip安装!!!
    参考https://pytorch.org/get-started/previous-versions/

  3. 安装torch-geometric相关依赖包
    (1)如果之前不确定安装的torch版本和对应的cuda,先通过如下语句查看:
    python -c "import torch; print(torch.__version__)"
    python -c "import torch; print(torch.version.cuda)"
    (2)然后安装相关包:
    pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html
    pip install torch-geometric
    此处的torch版本和cuda版本要与前面的对应!
    参考https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html

你可能感兴趣的:(python,pytorch,深度学习)