【Python】安装torch_geometric、以及torch_scatter系列安装

【Python】安装torch_geometric、以及torch_scatter系列安装

1. 介绍

首先，我们要知道：

• 安装torch_geometric，需要同时安装torch-scatter，torch-sparse，torch-cluster，torch-spline-conv等库

因此，你如果只需要torch_scatter的话，就安装它就好了，但是如果要torch_geometric的话，就需要全都下载了。

2. 方法

2.1 第一步：查看CUDA版本

import torch
print(torch.__version__)
print(torch.version.cuda)

注意：

• Pytorch和CUDA的版本一定要对应上，否则，即便安装了也会出错调用不了，一般情况下，不要安装最新版本Pytorch，因为CUDA和其他相关依赖库。
• 如果之前安装了其他版本的CUDA或者Pytorch，要卸载干净了再安装，如果没有卸载干净直接安装，系统就会有多个版本，就算用nvcc --version检测出来有安装该软件，也无法正常调用CUDA。

2.2 第二步：安装

## 先安装这四个
pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html

## 最后安装torch-geometric
pip install torch-geometric

其中 {TORCH} 替 换 为 P y T o r c h 的 版 本 号 ，{CUDA} 替换为上面查询得到的CUDA版本号。CUDA 版本号格式 (cpu, cu92, cu101, cu102, cu110, cu111)。例如对于PyTorch 2.0.0 and CUDA 11.7，安装命令如下：

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-2.0.0+cu117.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-2.0.0+cu117.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-2.0.0+cu117.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-2.0.0+cu117.html
pip install torch-geometric

2.3 第三步：检测是否安装成功

$ python
Python 3.7.0 (default, Jun 28 2018, 13:15:42)
[GCC 7.2.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> from torch_geometric.data import DataLoader
>>>

• 没有出现错误提示，即成功。

3. 参考

【1】https://blog.csdn.net/qq_41800917/article/details/120282841