python3.7对应pytorch版本_python3.7+pytorch1.7.1+cuda10.2环境安装torch_geometric

1、官网上查看自己的环境对应的torch_geometric版本

(1)github网址：https://github.com/rusty1s/pytorch_geometric

(2)打开网站之后，能在网页上看到  “Installation”，我用到的版本如下图：

2、安装torch_geometric的命令

我的电脑配置是cuda10.2,对应的是cu102(如果不是这个环境，可以更改下面命令中加粗的内容)，在命令行中运行如下代码即可：

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

3、是否需要其他依赖包？

在运行上述命令的过程中，我发现系统会自动下载一些包，通过查看原项目，可以看到有一个setup.py文件，打开后可以看到下图，numpy,  tqdm等都是本项目的依赖包。如果之前没有下载过这些包也不用担