今天真的是吐血,一定要把这个分享出去,让更多的人少走弯路。很少有人分享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“,
我确定Cuda版本的方法是:用everything软件搜索“cuda”关键字,在anaconda安装目录下发现如下图所示文件夹,从而确定cuda版本
题外话:
everything是一个非常方便且高效的文件搜索小工具,安装包可自取
链接:https://pan.baidu.com/s/18r4NGTTbchrKvBCM7bjxKA
提取码:vqja
四、安装的完成命令
则完整命令为(以Python3.7、pytorch1.7.1、cuda101为例):
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
不报错就成功了 !