跑深度学习模型Ⅲ：正确安装与torch版本对应的其他torch包

pytorch的正确安装可以回看我前面的博客跑深度学习模型Ⅱ：一文安装正确pytorch及dgl-CSDN博客

这篇博客将安装torch_grometric，torch_scatter, torch_sparse, torch_cluster库

1. 查看自己的torch版本

进入cmd 切换到要用的python环境中，输入以下命令

python

import torch

print(torch.__version__)

会返回你的pytorch版本，我的版本为2.2.1

2. 去官网找安装命令

进入官网Installation — pytorch_geometric documentation (pytorch-geometric.readthedocs.io)

在左下角选择版本，点开选择，直到找到图中标3这的pytorch版本 与你的pytorch版本一致。

再选择与你的cuda版本一致的。我的torch是2.2.1 cuda 是11.8。

（cuda的安装使用可以看我前面写的博客跑深度学习模型I：一文正确使用CUDA-CSDN博客）

3. 安装

pip install torch_geometric

pip install torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.2.0+cu118.html

将这些命令输入到python虚拟环境中进行安装即可。