PyTorch Geometric:PyTorch 的几何深度学习扩展库-开源

它由来自各种已发表论文的各种深度学习图和其他不规则结构的方法组成，也称为几何深度学习。 此外，它包含一个易于使用的小批量加载器，用于许多小型和单个巨型图、大量通用基准数据集（基于简单的接口创建自己的）和有用的转换，均用于学习任意图形以及 3D 网格或点云。 我们已经将 PyTorch Geometric 的很多功能外包给其他包，这些包需要额外安装。 这些软件包带有自己的基于 C++/CUDA 扩展的 CPU 和 GPU 内核实现。 我们不建议在您的系统 python 上以 root 用户身份安装。 请设置 Anaconda/Miniconda 环境或创建 Docker 镜像。 我们为所有主要的 OS/PyTorch/CUDA 组合提供 pip 轮。