最近在考虑用GNN,于是首先安装torch_geometric库。。。遇到一些小坑。。。供大家参考
一、查看环境
- 打开虚拟环境
conda activate work
- 查看torch和cuda版本
import torch
print(torch.__version__)
我这里的版本是1.10+11.1
3. 找对应版本的torch_geometric进行安装,注意这里用wheel文件安装,在安装geometric之前需要先按顺序安装如下内容:(点here下载必要依赖)
第一步安装pip install torch_scatter-2.0.9-cp37-cp37m-win_amd64.whl
第二步安装pip install torch_sparse-0.6.13-cp37-cp37m-win_amd64.whl
第三步安装pip install torch_cluster-1.6.0-cp37-cp37m-win_amd64.whl
第四步安装pip install torch_spline_conv-1.2.1-cp37-cp37m-win_amd64.whl
第五步安装pip install torch_geometric
- 在pycharm中输入import torch_geometric,没有显示报错即为安装成功。
报错解决
报错一:DLL load failed: 找不到指定的程序。
和我之前安装sklearn时候报的错一毛一样!
就不该动numpy
numpy是用wheel文件安装的,但是这次提示我升级,我直接用的pip install,导致出错(据说是由于在线pip install numpy会没有mkl),于是又找到wheel文件卸载numpy重新安装numpy,报错解决!
报错二
原因是因为PYG 2.3 requires that at least Pytorch 1.12 is installed.
解决方案
直接升级torch或者降级geometric就好了
后续写:介绍geometric
官方API:https://pytorch-geometric.readthedocs.io/en/latest/