安装

  • Python version >= 3.7

  • PyTorch version >= 1.7.1

请按照此处的说明安装 PyTorch

安装 PyTorch 后，可以使用 pip命令安装 cogdl，如下所示：

pip install cogdl

或者Install from source via：

pip install git+https://github.com/thudm/cogdl.git

或者clone仓库并使用以下命令进行安装：

git clone git@github.com:THUDM/cogdl.git
cd cogdl
pip install -e .

如果您想使用 PyTorch Geometric (PyG) 中的模块，您可以按照此处的说明安装 PyTorch Geometric