python 导入  py python导入pytorch_官网

pytorch_geometric安装方法

方法一

官方链接:installation Github:https://github.com/rusty1s/pytorch_geometric

方法二

  1. 查看自己主机的cuda版本已经安装好的pytorch的cuda版本
    nvcc --versioncat /usr/local/cuda/version.txtpython -c "import torch; print(torch.version.cuda)"
  2. python 导入  py python导入pytorch_python_02


2.安装cuda对应的pytorch

不建议pip安装,可以直接到官网找相应的版本下载

python 导入  py python导入pytorch_github_03


版本号和系统对应

  1. 安装torch_geometric
    官网找到合适的版本下载

    然后
    pip install torch-geometric

以下为我安装的文件目录

python 导入  py python导入pytorch_github_04

最后就可以运行GCN啦~