如何安装 torch_geometric？

知行合一，至于至善

已于 2024-09-17 19:38:01 修改

阅读量1.1w

点赞数 23

文章标签： pytorch 深度学习 python

于 2021-09-14 13:57:02 首次发布

本文链接：https://blog.csdn.net/qq_41800917/article/details/120282841

版权

torch_geometric安装官网：
https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html

1. 查看CUDA版本

方法一：直接在终端检查版本，输入命令nvidia-smi，如图所示：

基于系统终端查看CUDA版本
方法二：通过命令nvcc --version查看，前提是电脑上安装了CUDA Toolkit。如图所示：
在这里插入图片描述
方法三：基于控制面板查看

搜索并打开控制面板

在这里插入图片描述

打开系统信息
打开系统信息，并选择"组件"标签

在这里插入图片描述

如果没有安装cuda,参考以下网址安装：
https://blog.csdn.net/ashome123/article/details/105822040

2. 安装PyTorch

PyTorch官网：https://pytorch.org/
在这里插入图片描述

查看PyTorch版本

$ python -c "import torch; print(torch.__version__)"
>>> 1.8.0
print(torch.__version__)

如果没有安装pytorch，则去官网安装，网址如下：https://pytorch.org/get-started/locally/
whl文件：https://download.pytorch.org/whl/torch/

注意点：切记！切记！切记！

Pytorch和CUDA的版本一定要对应上；否则，即便安装了也会出错调用不了。一般情况下，不要安装最新版本Pytorch，因为CUDA和其他相关依赖库版本（安装torch_geometric，同时需要安装torch-scatter，torch-sparse，torch-cluster，torch-spline-conv等库）会跟不上，没有与之对应的版本，所以安装之前一定要核对好Pytorch和CUDA的版本是否对应上。
如果之前安装了其他版本的CUDA或者Pytorch，一定要卸载干净了再安装。如果没有卸载干净直接安装，系统就会有多个版本，就算用nvcc --version检测出来有安装该软件，也无法正常调用CUDA。

3. 安装 torch-geometric

参考官网：https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html
torch-geometric依赖库whl文件下载网址: https://data.pyg.org/whl/torch-1.8.0%2Bcu111.html

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
pip install torch_geometric==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

其中，${TORCH}替换为PyTorch的版本号，${CUDA}替换为上面查询得到的CUDA版本号。CUDA 版本号格式 (cpu, cu92, cu101, cu102, cu110, cu111)。例如，对于 PyTorch 1.8.0/1.8.1 and CUDA 11.1，安装命令如下：

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-cluster -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html
pip install torch_geometric==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

对于 PyTorch 1.7.0/1.7.1 and CUDA 10.2，安装命令如下：

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.0+cu102.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.7.0+cu102.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.7.0+cu102.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.7.0+cu102.html
pip install torch_geometric==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

4. 检测是否安装成功

$ 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
>>>

若没报错，则安装成功。但我的没有直接安装成功，而是出现了 5.可能出现的错误。

5. 可能出现的错误

按照以上安装步骤完成安装以后，还可能出现‘No module named 'torch_sparse'报错。
那么，可以进入torch_sparse官网找对应的cuda、pytorch版本安装。
torch_sparse官网：https://github.com/rusty1s/pytorch_sparse
我的是PyTorch 1.8.0/1.8.1
To install the binaries for PyTorch 1.8.0 and 1.8.1, simply run

pip install torch-scatter torch-sparse -f https://data.pyg.org/whl/torch-1.8.0+${CUDA}.html