torch-scatter -cluster -spline -geometric 安装

今天真的是吐血，一定要把这个分享出去，让更多的人少走弯路。很少有人分享torch-scatter -cluster -spline -geometric 安装，找淘宝店给我安装，他竟然说不会！我只能自己寻找方法了！

首先，相信很多人看过这个教程：https://github.com/rusty1s/pytorch_geometric#pytorch-170

我来排一排这个教程的坑，就可以安装成功了。

一、环境变量

一定要切换到想要安装库的环境变量中

conda env list          # 查看变量列表

activate 变量名        # 激活要安装库的变量

一、pytorch版本的选择

如果打开下载地址的上一级目录：https://pytorch-geometric.com/whl/，可以看到pytorch版本（注：没有1.7.1或1.6.**等版本的安装链接，如果已安装的pytorch是1.7.**版本，安装链接和1.7.0一样）

二、cuda的选择

与电脑已安装版本对应，查看版本后，再选择安装链接。

另，以cu101为例，教程中 “${CUDA}” 改为 ”cu101“，

则完整命令为（以1.7.1pytorch、cu101为例）：

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

三、安装完成测试

在cmd中，进入python环境：

(py37) C:\Users\xiaozhe>python
Python 3.7.9 (default, Aug 31 2020, 17:10:11) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.

 倒入库（注：库的名字中“-”改为下划线“_"才能成功导入）：

>>> import torch_scatter
>>> import torch_sparse
>>> import torch_cluster
>>> import torch_spline_conv
>>> import torch_geometric

不报错就成功了 ！

标签：1.7,cu101,torch,cluster,pytorch,geometric,安装,spline
来源： https://blog.csdn.net/weixin_43217427/article/details/114155502