最近做GCN需要引入torch-geometic这个包,花了一整天时间踩坑才安装好环境!!!!!给大家一次性把坑说明白!
安装小组件
仅仅安装好这个包的话是会报错的,在这之前要先安装好以下四个组件:torch-cluster
torch-scatter
torch-sparse
torch-spline-conv
但是这四个组件直接用pip或者conda是安装不上去的,所以需要先在以下网站下载whl文件到本地
https://pytorch-geometric.com/whl/torch-1.7.1.html
在这一步我就踩了一个大坑,就是这个网站后面的torch-1.7.1是不能直接套用的,一定要遵循自己的torch版本!!!!!!!加粗提醒!!!!!!!按照自己的torch版本去修改网站最后的torch-xxx再下载whl文件
然后再`pip install (直接复制whl文件的完整地址)
如何选择版本号
打开网站后会看到很多文件进行下载,每一次选择whl文件都请擦亮自己的眼睛:
1:版本号。
我是按照源代码的requirement文件的要求下载的,其实只要不下载最新的版本应该都没太大问题
2:cpu/gpu对应型号。一定要看准确!查看cuda版本的方法自己搜一下!
3:python解释器对应的版本号,我是py37,一般好像是用py38
4:操作系统:mac/window/linux
上面四个每一个都不能错!!!我的py版本和cpu混着错,每次都要检查很久排查错误呜呜呜呜呜,大家争取一次到位
安装torch-geometic
这个可以直接pip install torch-geometic==版本号
以上就是完整安装程序了!祝大家好运!!!