Pytorch中安装 torch_geometric 详细图文操作（全）

前言

如果已经安装了pytorch版本，对应跳过步骤一，直接安装步骤二即可

注意：依赖包的配置需要根据pytorch版本号以及cuda的版本号，版本号要适配，否则部分函数无法使用

1. 安装pytorch

基于代码安装适配的pytorch版本

通过官网进行安装：pytorch下载官网，此处版本根据代码需要，具体格式如下

• 命令如下（linux）：conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cudatoolkit=10.2 -c pytorch

其他操作系统以及版本号需要查看下官网，具体安装格式如上

2. 安装torch_geometric

在执行命令的时候发现缺失这个torch_geometric包

通过pip install 安装对应的包：

再次执行该代码，发现还是会报错：ModuleNotFoundError: No module named 'torch_sparse'

对应还需要如下四个依赖包配置（必须版本适配，否则会出现部分函数无法使用）：

3. 安装依赖包

本身我是装在虚拟环境下，通过conda list也可查看到我的配置：

通过 pytorch-geometric官网链接 下载额外的四个依赖包结合pytorch版本）：

命令如下（无先后顺序）：
（注意：不同pytorch以及cud版本不一样，选择的依赖包不一样，通过上面的官网查看，只需替代链接即可）

• 下载cluster：wget https://data.pyg.org/whl/torch-1.8.0%2Bcu102/torch_cluster-1.5.9-cp38-cp38-linux_x86_64.whl
• 下载scatter：wget https://data.pyg.org/whl/torch-1.8.0%2Bcu102/torch_scatter-2.0.8-cp38-cp38-linux_x86_64.whl
• 下载sparse：wget https://data.pyg.org/whl/torch-1.8.0%2Bcu102/torch_sparse-0.6.9-cp38-cp38-linux_x86_64.whl
• 下载spline：wget https://data.pyg.org/whl/torch-1.8.0%2Bcu102/torch_spline_conv-1.2.1-cp38-cp38-linux_x86_64.whl

下载之后，截图如下：

对应安装上面四个压缩包：pip install torch_*

安装完成看到success即已成功安装该包