pytorch_geometric是一个图神经网络的库。如果直接通过pip安装在运行时会显示依赖库(如torch-sparse)没安装。
在安装这些依赖库通过pip安装不上。
需要去官网 https://pytorch-geometric.com/whl/ 下载对应pytorch和cuda版本的依赖库。
或者用这样的代码进行安装(pytorch和cuda版本需要修改)
!pip install torch-geometric \
torch-sparse \
torch-scatter \
torch-cluster \
-f https://pytorch-geometric.com/whl/torch-1.8.0+cu101.html